diff --git a/config.py b/config.py
index cea5095..11d39b0 100644
--- a/config.py
+++ b/config.py
@@ -8,4 +8,4 @@ class Config(TypedDict):
     name: str
     project_dir: str
     end_date: str
-    openframe: bool
+    top_level_macro: str
diff --git a/shuttle.py b/shuttle.py
index c0a61fb..713c340 100644
--- a/shuttle.py
+++ b/shuttle.py
@@ -31,10 +31,7 @@ def __init__(self, config: Config, projects: List[Project], modules_yaml_name: s
         self.script_dir = os.path.dirname(os.path.realpath(__file__))
         self.modules_yaml_name = modules_yaml_name
         self.mux_config_yaml_name = os.environ.get("TT_CONFIG", "sky130.yaml")
-        if config.get("openframe", False):
-            self.tt_top_macro = "openframe_project_wrapper"
-        else:
-            self.tt_top_macro = "user_project_wrapper"
+        self.tt_top_macro = config.get("top_level_macro", "openframe_project_wrapper")
 
         self.read_mux_config_file()
         total_rows: int = self.mux_config["tt"]["grid"]["y"]