diff --git a/dagster_sqlmesh/resource.py b/dagster_sqlmesh/resource.py
index 2220d75..cc0e389 100644
--- a/dagster_sqlmesh/resource.py
+++ b/dagster_sqlmesh/resource.py
@@ -256,4 +256,6 @@ def run(
                 yield from event_handler.process_events(mesh.context, event)
 
     def get_controller(self, log_override: t.Optional[logging.Logger] = None):
-        return SQLMeshController.setup(self.config, log_override=log_override)
+        return SQLMeshController.setup_with_config(
+            self.config, log_override=log_override
+        )
diff --git a/pyproject.toml b/pyproject.toml
index a804212..85abcbd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dagster-sqlmesh"
-version = "0.3.0"
+version = "0.3.1"
 description = ""
 authors = ["Reuven Gonzales <reuven@karibalabs.co>"]
 license = "Apache 2.0"