diff --git a/.github/workflows/unittests.yml b/.github/workflows/unittests.yml
index 937a2bf..aee2244 100644
--- a/.github/workflows/unittests.yml
+++ b/.github/workflows/unittests.yml
@@ -9,5 +9,5 @@ jobs:
     steps:
     - uses: actions/checkout@v2
     - run: cd docker && ./build.sh
-    - run: docker run --rm cryo-em:latest python -m unittest discover tests
+    - run: docker run --rm cryoem:latest python -m unittest discover tests
 
diff --git a/run.py b/run.py
index 793c31f..f8a70e1 100644
--- a/run.py
+++ b/run.py
@@ -43,8 +43,8 @@ def main():
     parser = create_parser()
     args = parser.parse_args()
     server = PluginServer()
-    host = os.environ['NTS_HOST']
-    port = int(os.environ['NTS_PORT'])
+    host = args.host or os.environ.get('NTS_HOST', None)
+    port = args.port or int(os.environ.get('NTS_PORT', 8888))
     name = "Cryo-EM"
     description = "Nanome plugin to load Cryo-EM maps and display them in Nanome as iso-surfaces"
     plugin_class = CryoEM