diff --git a/simulationworkflowschema/__init__.py b/simulationworkflowschema/__init__.py
index 4210c0b..719c0a7 100644
--- a/simulationworkflowschema/__init__.py
+++ b/simulationworkflowschema/__init__.py
@@ -94,7 +94,13 @@ def load_modules():
         'photon_polarization',
         'thermodynamics',
     ]
+    import simulationworkflowschema
 
+    for name in sub_modules:
+        sub_module = importlib.import_module(f'simulationworkflowschema.{name}')
+        for method in sub_module.__dict__:
+            if method in __all__:
+                setattr(simulationworkflowschema, method, sub_module.__dict__[method])
 
 class SimulationWorkflowSchemaEntryPoint(SchemaPackageEntryPoint):
     def load(self):