diff --git a/src/lbaf/IO/lbsGridStreamer.py b/src/lbaf/IO/lbsGridStreamer.py
index 62c33895..532ad572 100644
--- a/src/lbaf/IO/lbsGridStreamer.py
+++ b/src/lbaf/IO/lbsGridStreamer.py
@@ -1,6 +1,10 @@
 from typing import Optional
 
-import vtk
+try:
+    import vtk
+    using_vtk = True
+except ModuleNotFoundError:
+    using_vtk = False
 
 from ..Utils.lbsLogging import Logger, get_logger