diff --git a/portainer.py b/portainer.py
index abd35ae..165cb13 100755
--- a/portainer.py
+++ b/portainer.py
@@ -13,7 +13,7 @@ import json
 import argparse
 import hvac
 from tabulate import tabulate
-from port import PortainerApi
+import port
 from prompt_toolkit import prompt
 from prompt_toolkit.completion import WordCompleter
 from prompt_toolkit.shortcuts import checkboxlist_dialog
@@ -425,7 +425,7 @@ if __name__ == "__main__":
         
         os.system("cls" if os.name == "nt" else "clear")
     # Example: list endpoints
-    por = PortainerApi(cur_config["PORTAINER_SITE"], args)
+    por = port.PortainerApi(cur_config["PORTAINER_SITE"], args)
     por.set_defaults(cur_config)
     if args.debug:
         por._debug = True