diff --git a/pyproject.toml b/pyproject.toml
index 57d0118..33792b6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,7 +17,7 @@ namespaces = true
 # ----------------------------------------- Project Metadata -------------------------------------
 #
 [project]
-version = "0.0.0.dev348"
+version = "0.0.0.dev349"
 name = "ControlMan"
 dependencies = [
     "packaging >= 23.2, < 24",
diff --git a/src/controlman/file_gen/python.py b/src/controlman/file_gen/python.py
index c927f33..b14678d 100644
--- a/src/controlman/file_gen/python.py
+++ b/src/controlman/file_gen/python.py
@@ -141,7 +141,7 @@ def python_files(self) -> list[DynamicFile]:
         for filepath in abs_path.glob("**/*.py"):
             file_content = filepath.read_text()
             if mapping:
-                file_content = _pysyntax.modify.imports(module_content=file_content, mapping=mapping)
+                file_content = _pysyntax.modify.imports(code=file_content, mapping=mapping)
             if filepath in path_to_globs_map:
                 for config_id, file_config in path_to_globs_map[filepath]:
                     if "docstring" in file_config: