diff --git a/src/codegen/cli/cli.py b/src/codegen/cli/cli.py
index 9ac7f69f4..8910f615a 100644
--- a/src/codegen/cli/cli.py
+++ b/src/codegen/cli/cli.py
@@ -1,7 +1,7 @@
 import rich_click as click
 from rich.traceback import install
 
-from codegen.cli.commands.agent.main import agent_command
+# Removed reference to non-existent agent module
 from codegen.cli.commands.config.main import config_command
 from codegen.cli.commands.create.main import create_command
 from codegen.cli.commands.deploy.main import deploy_command
@@ -32,7 +32,7 @@ def main():
 
 
 # Wrap commands with error handler
-main.add_command(agent_command)
+# Removed reference to non-existent agent_command
 main.add_command(init_command)
 main.add_command(logout_command)
 main.add_command(login_command)