diff --git a/rl_coach/coach.py b/rl_coach/coach.py
index 2542eb4..7c0274f 100644
--- a/rl_coach/coach.py
+++ b/rl_coach/coach.py
@@ -579,7 +579,7 @@ class CoachLauncher(object):
             return
 
         if args.distributed_coach and args.distributed_coach_run_type == RunType.ORCHESTRATOR:
-            handle_distributed_coach_orchestrator(graph_manager, args)
+            handle_distributed_coach_orchestrator(args)
             return
 
         # Single-threaded runs
diff --git a/rl_coach/memories/backend/redis.py b/rl_coach/memories/backend/redis.py
index 9a94880..b6010a0 100644
--- a/rl_coach/memories/backend/redis.py
+++ b/rl_coach/memories/backend/redis.py
@@ -113,6 +113,8 @@ class RedisPubSubBackend(MemoryBackend):
     def undeploy(self):
         if self.params.deployed:
             return
+
+        from kubernetes import client
         api_client = client.AppsV1Api()
         delete_options = client.V1DeleteOptions()
         try: