diff --git a/main.py b/main.py
index 3e50d64..d725ea7 100644
--- a/main.py
+++ b/main.py
@@ -3,7 +3,7 @@ from src import CodexAgent, CodexAgentConfig
 codex_agent = CodexAgent(CodexAgentConfig())
 
 def main():
-    codex_agent.push_to_hub("darinkishore/codex-agent", with_code=True)
+    codex_agent.push_to_hub("darinkishore/codex-agent", with_code=True, commit_message="Do not call forward directly")
 
 if __name__ == "__main__":
     main()
\ No newline at end of file
diff --git a/src/index.py b/src/index.py
index 9a78784..237002f 100644
--- a/src/index.py
+++ b/src/index.py
@@ -34,4 +34,4 @@ class CodexAgent(PrecompiledAgent):
         )
     
     def forward(self, **kwargs) -> Prediction:
-        return self.codex_module.forward(**kwargs)
\ No newline at end of file
+        return self.codex_module(**kwargs)
\ No newline at end of file