diff --git a/port.py b/port.py
index 919f124..6a9fc44 100644
--- a/port.py
+++ b/port.py
@@ -532,10 +532,10 @@ class Portainer:
         if not args.autostart:
             time.sleep(120)
             cont = []
-            for c in self.all_data["containers"][endpoint]:
-                if stack == c or stack == "all":
-                    cont += self.all_data["containers"][endpoint][c]
-            self.stop_containers(endpoint, cont)
+            for c in self.all_data["containers"][args.endpoint_id]:
+                if args.stack == c or args.stack == "all":
+                    cont += self.all_data["containers"][args.endpoint_id][c]
+            self.stop_containers(args.endpoint_id, cont)
 
     def get_endpoints(self, timeout=10):
         '''Get a list of all endpoints.'''
diff --git a/requirements.txt b/requirements.txt
index 5e7adcd..747db26 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ tabulate
 # Other dev tools
 flake8
 pylint
-black
\ No newline at end of file
+black
+docker
\ No newline at end of file