diff --git a/docker/Dockerfile.finn_dev b/docker/Dockerfile.finn_dev
index 46b3ffb255f6e3d4ff6e92d83b285e53be4beeeb..35bc316b6951de5b69bfe611427b9b9321f59347 100644
--- a/docker/Dockerfile.finn_dev
+++ b/docker/Dockerfile.finn_dev
@@ -54,7 +54,7 @@ RUN pip install sphinx==3.1.2
 RUN pip install sphinx_rtd_theme==0.5.0
 RUN pip install pytest-xdist==2.0.0
 RUN pip install pytest-parallel==0.1.0
-RUN pip install netron
+RUN pip install netron>=4.7.9
 RUN pip install -e git+https://github.com/fbcotter/dataset_loading.git@0.0.4#egg=dataset_loading
 
 # switch user
diff --git a/src/finn/util/visualization.py b/src/finn/util/visualization.py
index 9f0e9e8d773dd3cb4521d44db486a3f916b86ff7..3eb7e55e307c380ecc6712ff4d0c74577a9e7a43 100644
--- a/src/finn/util/visualization.py
+++ b/src/finn/util/visualization.py
@@ -34,7 +34,6 @@ from IPython.display import IFrame
 def showSrc(what):
     print("".join(inspect.getsourcelines(what)[0]))
 
-
 def showInNetron(model_filename):
-    netron.start(model_filename, port=8081, host="0.0.0.0")
-    return IFrame(src="http://0.0.0.0:8081/", width="100%", height=400)
+    netron.start(model_filename, address=("0.0.0.0", 8081))
+    return IFrame(src="http://0.0.0.0:8081/", width="100%", height=400)
\ No newline at end of file