diff --git a/autogpt/commands/__init__.py b/autogpt/commands/__init__.py
index 939e66ed..861da27d 100644
--- a/autogpt/commands/__init__.py
+++ b/autogpt/commands/__init__.py
@@ -5,4 +5,5 @@ COMMAND_CATEGORIES = [
     "autogpt.commands.web_search",
     "autogpt.commands.web_selenium",
     "autogpt.commands.system",
+    "autogpt.commands.image_gen",
 ]