diff --git a/lib/modules/ai_helper/semantic_categorizer.rb b/lib/modules/ai_helper/semantic_categorizer.rb
index 6a19075b..b759c6c3 100644
--- a/lib/modules/ai_helper/semantic_categorizer.rb
+++ b/lib/modules/ai_helper/semantic_categorizer.rb
@@ -2,9 +2,9 @@
 module DiscourseAi
   module AiHelper
     class SemanticCategorizer
-      def initialize(text, user)
+      def initialize(input, user)
         @user = user
-        @text = text
+        @text = input[:text]
       end
 
       def categories