diff --git a/module/ai/control.php b/module/ai/control.php
index 34522200dd..6056bc456f 100644
--- a/module/ai/control.php
+++ b/module/ai/control.php
@@ -302,7 +302,7 @@ class ai extends control
             return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSelectDataSource', "promptID=$promptID") . '#app=admin'));
         }
 
-        $this->view->activeDataSource = empty($prompt->module) ? 'my' : $prompt->module;
+        $this->view->activeDataSource = empty($prompt->module) ? current(array_keys($this->config->ai->dataSource)) : $prompt->module;
         $this->view->dataSource       = $this->config->ai->dataSource;
         $this->view->prompt           = $prompt;
         $this->view->promptID         = $promptID;