diff --git a/module/story/ui/create.html.php b/module/story/ui/create.html.php
index 19e7cc5400..974a8ab681 100644
--- a/module/story/ui/create.html.php
+++ b/module/story/ui/create.html.php
@@ -88,7 +88,7 @@ formPanel
                     ),
                     a
                     (
-                        setData(array('on' => 'click', 'call' => 'loadProductModules', 'params' => $productID)),
+                        set('onclick', "loadProductModules({$productID})"),
                         icon('refresh')
                     )
                 ) : null
diff --git a/module/tree/control.php b/module/tree/control.php
index ec1d69940e..76762d2f17 100644
--- a/module/tree/control.php
+++ b/module/tree/control.php
@@ -237,6 +237,7 @@ class tree extends control
             if(dao::isError()) return $this->sendError(dao::getError());
 
             if($this->viewType == 'json') return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'idList' => $moduleIDList));
+            if(helper::isAjaxRequest('modal')) return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'closeModal' => true));
             return $this->sendSuccess(array('load' => true));
         }
     }