diff --git a/module/ai/js/browseminiprogram.ui.js b/module/ai/js/browseminiprogram.ui.js
index 8ddf2e253b..db20474bb9 100644
--- a/module/ai/js/browseminiprogram.ui.js
+++ b/module/ai/js/browseminiprogram.ui.js
@@ -386,6 +386,8 @@ function createServerNotification(content)
 
 $(function()
 {
+    $('#heading .toolbar-item').attr('data-app', 'ai');
+
     $('#to-language-model').prop('href', $.createLink('ai', 'models'));
     $('#reload-current').on('click', () =>
     {
diff --git a/module/ai/js/square.ui.js b/module/ai/js/square.ui.js
index ee104baa38..ef58bd7e6f 100644
--- a/module/ai/js/square.ui.js
+++ b/module/ai/js/square.ui.js
@@ -21,3 +21,8 @@ window.aiSquare.handleStarBtnClick = function(e)
         }
     }, 'json');
 };
+
+$(function()
+{
+    $('#heading .toolbar-item').attr('data-app', 'ai');
+});