diff --git a/src/codegen/sdk/core/codebase.py b/src/codegen/sdk/core/codebase.py
index a1dd9662d..ee14c9f99 100644
--- a/src/codegen/sdk/core/codebase.py
+++ b/src/codegen/sdk/core/codebase.py
@@ -391,7 +391,7 @@ def types(self) -> list[TTypeAlias]:
     # EXTERNAL API
     ####################################################################################################################
 
-    def create_file(self, filepath: str, content: str = "", sync: bool = True) -> File:
+    def create_file(self, filepath: str, content: str = "", sync: bool = True) -> TSourceFile:
         """Creates a new file in the codebase with specified content.
 
         Args: