diff --git a/dpsynth/discrete_mechanisms/__init__.py b/dpsynth/discrete_mechanisms/__init__.py
index 0c47d36..c93a0df 100644
--- a/dpsynth/discrete_mechanisms/__init__.py
+++ b/dpsynth/discrete_mechanisms/__init__.py
@@ -23,3 +23,4 @@
 from dpsynth.discrete_mechanisms.direct import DirectConfig
 from dpsynth.discrete_mechanisms.independent import IndependentConfig
 from dpsynth.discrete_mechanisms.mst import MSTConfig
+from dpsynth.discrete_mechanisms.swift import SWIFTConfig
diff --git a/dpsynth/discrete_mechanisms/api.py b/dpsynth/discrete_mechanisms/api.py
index 15a05bf..f86b233 100644
--- a/dpsynth/discrete_mechanisms/api.py
+++ b/dpsynth/discrete_mechanisms/api.py
@@ -24,6 +24,7 @@
 from . import direct
 from . import independent
 from . import mst
+from . import swift
 
 
 class DiscreteMechanismConfig(Protocol):
@@ -65,6 +66,8 @@ def run_mechanism(
     run_mechanism_fn = direct.run_mechanism
   elif isinstance(config, independent.IndependentConfig):
     run_mechanism_fn = independent.run_mechanism
+  elif isinstance(config, independent.SWIFTConfig):
+    run_mechanism_fn = swift.run_mechanism
   else:
     raise ValueError(f'Unknown mechanism: {type(config)}')