diff --git a/src/__init__.py b/src/__init__.py
index 5c4f6f1..7ee26e1 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -17,7 +17,7 @@ class Pelato:
         
     def generate(self, project_dir):
         print(f"Generating Go code for project {project_dir}")
-        code_generator.generate(project_dir)
+        code_generator.generate(project_dir, self.registry_url)
         
     def build(self, project_dir):
         print(f"Building WASM component for project {project_dir}")