diff --git a/module/ai/config.php b/module/ai/config.php
index 669f4315f4..d2d529ca15 100644
--- a/module/ai/config.php
+++ b/module/ai/config.php
@@ -197,12 +197,50 @@ $config->ai->menuPrint->locations['doc']['view'] = (object)array(
 
 $config->ai->injectAuditButton = new stdclass();
 $config->ai->injectAuditButton->locations = array();
-$config->ai->injectAuditButton->locations['task']['edit']['toolbar'] = (object)array(
-    'targetContainer' => '#mainContent .main-header',
-    'injectMethod'    => 'append',
-    'class'           => 'pull-right btn-toolbar',
+$config->ai->injectAuditButton->locations['task']['edit'] = array(
+    'toolbar' => (object)array(
+        'targetContainer' => '#mainContent .main-header',
+        'injectMethod'    => 'append',
+        'class'           => 'pull-right btn-toolbar',
+    ),
+    'action' => (object)array(
+        'targetContainer' => '#mainContent .form-actions',
+        'injectMethod'    => 'html',
+    )
 );
-$config->ai->injectAuditButton->locations['task']['edit']['action'] = (object)array(
-    'targetContainer' => '#mainContent .form-actions',
-    'injectMethod'    => 'html',
+
+$config->ai->injectAuditButton->locations['bug']['create'] = array(
+    'toolbar' => (object)array(
+        'targetContainer' => '#mainContent .main-header .btn-toolbar',
+        'injectMethod'    => 'prepend',
+    ),
+    'action' => (object)array(
+        'targetContainer' => '#mainContent .form-actions',
+        'injectMethod'    => 'html',
+    )
 );
+
+$config->ai->injectAuditButton->locations['doc']['edit'] = array(
+    'toolbar' => (object)array(
+        'targetContainer' => '#mainContent #headerBox .btn-tools',
+        'injectMethod'    => 'prepend',
+    ),
+    'action' => (object)array(
+        'targetContainer' => '#mainContent #headerBox .btn-tools',
+        'injectMethod'    => 'html',
+    )
+);
+
+$config->ai->injectAuditButton->locations['productplan']['create'] = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['productplan']['edit']   = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['programplan']['create'] = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['bug']['edit']           = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['story']['change']       = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['testcase']['edit']      = $config->ai->injectAuditButton->locations['task']['edit'];
+$config->ai->injectAuditButton->locations['testreport']['create']  = $config->ai->injectAuditButton->locations['task']['edit'];
+
+$config->ai->injectAuditButton->locations['story']['batchcreate'] = $config->ai->injectAuditButton->locations['bug']['create'];
+$config->ai->injectAuditButton->locations['story']['create']      = $config->ai->injectAuditButton->locations['bug']['create'];
+$config->ai->injectAuditButton->locations['task']['batchcreate']  = $config->ai->injectAuditButton->locations['bug']['create'];
+$config->ai->injectAuditButton->locations['task']['create']       = $config->ai->injectAuditButton->locations['bug']['create'];
+$config->ai->injectAuditButton->locations['testcase']['create']   = $config->ai->injectAuditButton->locations['bug']['create'];
diff --git a/module/bug/view/create.html.php b/module/bug/view/create.html.php
index 4ffec1b105..d5dee015d5 100644
--- a/module/bug/view/create.html.php
+++ b/module/bug/view/create.html.php
@@ -14,7 +14,6 @@
 include '../../common/view/header.html.php';
 include '../../common/view/kindeditor.html.php';
 include '../../common/view/datepicker.html.php';
-include '../../ai/view/inputinject.html.php';
 js::set('holders', $lang->bug->placeholder);
 js::set('page', 'create');
 js::set('createRelease', $lang->release->create);
@@ -379,4 +378,5 @@ $('#osBox').next('.table-col').remove();
 $('#typeBox').closest('tr').append('<td>' + browser + '</td>');
 </script>
 <?php endif;?>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/bug/view/edit.html.php b/module/bug/view/edit.html.php
index 038a078ef1..404294fd66 100644
--- a/module/bug/view/edit.html.php
+++ b/module/bug/view/edit.html.php
@@ -14,7 +14,6 @@
 include '../../common/view/header.html.php';
 include '../../common/view/datepicker.html.php';
 include '../../common/view/kindeditor.html.php';
-include '../../ai/view/inputinject.html.php';
 js::set('page'                   , 'edit');
 js::set('changeProductConfirmed' , false);
 js::set('changeExecutionConfirmed' , false);
@@ -329,4 +328,5 @@ if($this->app->tab == 'project')   js::set('objectID', $bug->project);
     </div>
   </form>
 </div>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/doc/view/edittexttype.html.php b/module/doc/view/edittexttype.html.php
index 4d4e0272cc..8e9968a89a 100644
--- a/module/doc/view/edittexttype.html.php
+++ b/module/doc/view/edittexttype.html.php
@@ -11,7 +11,6 @@
  */
 ?>
 <?php include '../../common/view/header.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php if($doc->contentType == 'html')     include '../../common/view/kindeditor.html.php';?>
 <?php if($doc->contentType == 'markdown') include '../../common/view/markdown.html.php';?>
 <style>
@@ -150,4 +149,5 @@ $(function()
 <?php js::set('draft', $doc->draft);?>
 <?php js::set('type', 'doc');?>
 <?php js::set('titleNotEmpty', sprintf($lang->error->notempty, $lang->doc->title));?>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/productplan/view/create.html.php b/module/productplan/view/create.html.php
index 9315bc712c..371d7607d5 100644
--- a/module/productplan/view/create.html.php
+++ b/module/productplan/view/create.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('weekend', $config->execution->weekend);?>
 <?php js::set('productID', $productID);?>
 <?php js::set('lastLang', $lang->productplan->last);?>
@@ -99,4 +98,5 @@
     </form>
   </div>
 </div>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/productplan/view/edit.html.php b/module/productplan/view/edit.html.php
index da6cb68752..b7b6dfa30c 100644
--- a/module/productplan/view/edit.html.php
+++ b/module/productplan/view/edit.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('weekend', $config->execution->weekend);?>
 <?php js::import($jsRoot . 'misc/date.js');?>
 <?php js::set('today', helper::today());?>
@@ -90,4 +89,5 @@
     </form>
   </div>
 </div>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/programplan/view/create.html.php b/module/programplan/view/create.html.php
index 8310dbea03..5df29f668f 100644
--- a/module/programplan/view/create.html.php
+++ b/module/programplan/view/create.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/sortable.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('browseType', $type);?>
 <style>.icon-help{margin-left: 3px;}</style>
 <div id='mainMenu' class='clearfix'>
@@ -284,4 +283,5 @@ var options = {
 }
 $('#planForm tbody.sortable').sortable(options);
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/story/view/batchcreate.html.php b/module/story/view/batchcreate.html.php
index 0ac6cf7334..576e48ca58 100755
--- a/module/story/view/batchcreate.html.php
+++ b/module/story/view/batchcreate.html.php
@@ -11,7 +11,6 @@
  */
 ?>
 <?php include './header.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php if($forceReview) $config->story->create->requiredFields .= ',review';?>
 <?php js::set('showFields', $showFields);?>
 <?php js::set('requiredFields', $config->story->create->requiredFields);?>
@@ -284,4 +283,5 @@ $(function()
 </script>
 <?php endif;?>
 <?php include '../../common/view/pastetext.html.php';?>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/story/view/change.html.php b/module/story/view/change.html.php
index b11bdeae34..58f832d483 100644
--- a/module/story/view/change.html.php
+++ b/module/story/view/change.html.php
@@ -11,7 +11,6 @@
  */
 ?>
 <?php include './header.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('lastReviewer', explode(',', $lastReviewer))?>
 <div id='mainContent' class='main-content'>
   <div class='center-block'>
@@ -115,4 +114,5 @@
 <?php js::set('storyType', $story->type);?>
 <?php js::set('rawModule', $this->app->rawModule);?>
 <?php js::set('page', $this->app->rawMethod);?>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/story/view/create.html.php b/module/story/view/create.html.php
index a70a8b9acd..0d7c35521c 100644
--- a/module/story/view/create.html.php
+++ b/module/story/view/create.html.php
@@ -11,7 +11,6 @@
  */
 ?>
 <?php include './header.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('page', 'create');?>
 <?php js::set('holders', $lang->story->placeholder); ?>
 <?php js::set('blockID', $blockID); ?>
@@ -426,4 +425,5 @@ setTimeout(() => {
 }, 600);
 
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/task/view/batchcreate.html.php b/module/task/view/batchcreate.html.php
index 86038f6bf1..6edc0a2e44 100755
--- a/module/task/view/batchcreate.html.php
+++ b/module/task/view/batchcreate.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/datepicker.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('taskConsumed', $taskConsumed);?>
 <?php js::set('addChildTask', $lang->task->addChildTask);?>
 <?php js::set('isonlybody', isonlybody())?>
@@ -345,4 +344,5 @@ $(function()
 </script>
 <?php endif;?>
 <?php include '../../common/view/pastetext.html.php';?>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/task/view/create.html.php b/module/task/view/create.html.php
index ea106e2b05..cf4f9167f3 100644
--- a/module/task/view/create.html.php
+++ b/module/task/view/create.html.php
@@ -13,7 +13,6 @@
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
 <?php include '../../common/view/sortable.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('toTaskList', !empty($task->id));?>
 <?php js::set('blockID', $blockID);?>
 <?php js::set('executionID', $execution->id);?>
@@ -355,4 +354,5 @@ foreach(explode(',', $config->task->create->requiredFields) as $field)
 <script>
 $(function(){parent.$('body.hide-modal-close').removeClass('hide-modal-close');})
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/testcase/view/create.html.php b/module/testcase/view/create.html.php
index b91e5db9c0..14e2558f51 100644
--- a/module/testcase/view/create.html.php
+++ b/module/testcase/view/create.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('page', 'create');?>
 <?php js::set('lblDelete', $lang->testcase->deleteStep);?>
 <?php js::set('lblBefore', $lang->testcase->insertBefore);?>
@@ -309,4 +308,5 @@ $(":checkbox[name^='auto']").on('click', function(){
     checkScript();
 });
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/testcase/view/edit.html.php b/module/testcase/view/edit.html.php
index 83fa33e9de..03c5b9385b 100644
--- a/module/testcase/view/edit.html.php
+++ b/module/testcase/view/edit.html.php
@@ -12,7 +12,6 @@
 ?>
 <?php include '../../common/view/header.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('page', 'edit');?>
 <?php js::set('lblDelete', $lang->testcase->deleteStep);?>
 <?php js::set('lblBefore', $lang->testcase->insertBefore);?>
@@ -354,4 +353,5 @@ $(":checkbox[name^='auto']").on('click', function(){
     checkScript();
 });
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/testreport/view/create.html.php b/module/testreport/view/create.html.php
index 46969e6093..adf6fae1ab 100644
--- a/module/testreport/view/create.html.php
+++ b/module/testreport/view/create.html.php
@@ -14,7 +14,6 @@
 <?php include '../../common/view/datepicker.html.php';?>
 <?php include '../../common/view/kindeditor.html.php';?>
 <?php include '../../common/view/chart.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 <?php js::set('objectType', $objectType);?>
 <div id='mainContent' class='main-content'>
   <div class='center-block'>
@@ -148,4 +147,5 @@ objectType = $("#objectType").val();
 extra      = '<?php echo $extra;?>';
 method     = 'create';
 </script>
+<?php include '../../ai/view/inputinject.html.php';?>
 <?php include '../../common/view/footer.html.php';?>
diff --git a/module/tree/view/browse.html.php b/module/tree/view/browse.html.php
index 5921391d1b..6277d0a731 100644
--- a/module/tree/view/browse.html.php
+++ b/module/tree/view/browse.html.php
@@ -11,7 +11,6 @@
  */
 ?>
 <?php include '../../common/view/header.html.php';?>
-<?php include '../../ai/view/inputinject.html.php';?>
 
 <?php if($viewType != 'story'):?>
 <style>
@@ -371,3 +370,4 @@ else
     include '../../common/view/footer.html.php';
 }
 ?>
+<?php include '../../ai/view/inputinject.html.php';?>