diff --git a/grants_tagger/streamlit_visualize.py b/grants_tagger/streamlit_visualize.py
index 567ea449..cd5a33f3 100644
--- a/grants_tagger/streamlit_visualize.py
+++ b/grants_tagger/streamlit_visualize.py
@@ -13,7 +13,7 @@
     )
     SHAP_IMPORTED = False
 
-from grants_tagger.models import MeshCNN
+from grants_tagger.models.mesh_cnn import MeshCNN
 from grants_tagger.predict import predict_tags