diff --git a/flipstreamviewer.py b/flipstreamviewer.py
index 060d771..d26d507 100644
--- a/flipstreamviewer.py
+++ b/flipstreamviewer.py
@@ -44,7 +44,7 @@ except:
     film = None
 
 try:
-    from custom_nodes.ComfyUI_TensorRT import TensorRTLoader
+    from custom_nodes.comfyui_tensorrt import TensorRTLoader
 except:
     TensorRTLoader = None