diff --git a/lib/ai_helper/assistant.rb b/lib/ai_helper/assistant.rb
index 03e6fa54..7c965fa8 100644
--- a/lib/ai_helper/assistant.rb
+++ b/lib/ai_helper/assistant.rb
@@ -141,7 +141,7 @@ module DiscourseAi
           %w[post]
         when "summarize"
           %w[post]
-        when "illustrate_posts"
+        when "illustrate_post"
           %w[composer]
         else
           %w[composer post]
diff --git a/lib/ai_helper/entry_point.rb b/lib/ai_helper/entry_point.rb
index 3ab9daa9..87ebca24 100644
--- a/lib/ai_helper/entry_point.rb
+++ b/lib/ai_helper/entry_point.rb
@@ -7,7 +7,7 @@ module DiscourseAi
           Rails.root.join("plugins", "discourse-ai", "db", "fixtures", "ai_helper"),
         )
 
-        additional_icons = %w[spell-check language]
+        additional_icons = %w[spell-check language images]
         additional_icons.each { |icon| plugin.register_svg_icon(icon) }
 
         plugin.on(:chat_thread_created) do |thread|