diff --git a/src/codegen/utilities/function_helpers.py b/src/codegen/utilities/function_helpers.py
index a54c7422..d3966955 100644
--- a/src/codegen/utilities/function_helpers.py
+++ b/src/codegen/utilities/function_helpers.py
@@ -40,7 +40,7 @@ def get_functions(metadata, class_name=""):
     all_functions = deepcopy(metadata["functions"])
     functions_metadata = []
     for function_name, function_data in all_functions.items():
-        if "python_codegen_method" in function_data or function_name in EXCLUDED_FUNCTIONS:
+        if function_data.get("python_codegen_method") or function_name in EXCLUDED_FUNCTIONS:
             continue
         if (
             "python_class_name" in all_functions[function_name]