diff --git a/slate/slate.py b/slate/slate.py
index 5e88607..b4a531a 100644
--- a/slate/slate.py
+++ b/slate/slate.py
@@ -215,7 +215,7 @@ class Slate():
         if self.consume(slurmC, 'pre_validate', True):
             Runner = self.runners[runnerName]
             runner = Runner(self, config)
-            runner.setup()
+            runner.setup(wandbC['job_type'])
 
         self._init_sweep(config)
         self.consume(config, 'wandb')
@@ -323,7 +323,7 @@ class Slate():
                         **wandbC
                     ) as run:
                         runner = Runner(self, runnerConf)
-                        runner.setup()
+                        runner.setup(wandbC['job_type'])
                         runner.run(run)
                 except wandb.errors.CommError as e:
                     retry -= 1
@@ -363,7 +363,7 @@ class Slate():
             self.deep_update(config, wandb.config)
             run.config = copy.deepcopy(config)
             runner = Runner(self, config)
-            runner.setup()
+            runner.setup(wandbC['job_type'])
             runner.run(run)
 
             if config != {}: