diff --git a/slate/slate.py b/slate/slate.py
index 4debec8..f882934 100644
--- a/slate/slate.py
+++ b/slate/slate.py
@@ -135,7 +135,7 @@ class Slate():
         ) as run:
             config = copy.deepcopy(orig_config)
             self.deep_update(config, wandb.config)
-            runner(run, config)
+            runner(self, run, config)
 
         assert config == {}, ('Config was not completely consumed: ', config)
 
@@ -171,7 +171,7 @@ class Slate():
             config=config,
             **wandbC
         ) as run:
-            runner(run, config)
+            runner(self, run, config)
 
         assert config == {}, ('Config was not completely consumed: ', config)