diff --git a/modules/ai_engine_chat/js/events.js b/modules/ai_engine_chat/js/events.js
index 1b5dd8a..0ed5e55 100644
--- a/modules/ai_engine_chat/js/events.js
+++ b/modules/ai_engine_chat/js/events.js
@@ -8,6 +8,7 @@
 document.addEventListener("DOMContentLoaded", function () {
   var launchLinks = document.querySelectorAll('a[href="#launch-chat"]');
   launchLinks.forEach(function (link) {
+    link.classList.add("ai-chatbot");
     link.addEventListener("click", function (event) {
       event.preventDefault();
       // Trigger a click on the button with id "launch-chat-modal".