diff --git a/src/scoreBasedDenoising/__init__.py b/src/scoreBasedDenoising/__init__.py
index 2dcd948..900803c 100644
--- a/src/scoreBasedDenoising/__init__.py
+++ b/src/scoreBasedDenoising/__init__.py
@@ -109,7 +109,8 @@ def getModelPath(self):
             if not path.exists():
                 raise FileNotFoundError(f"{path} does not exist.")
             return Path(self.model_path)
-        modelDir = impRes.files("graphite.pretrained_models.denoiser")
+        # modelDir = impRes.files("graphite.pretrained_models.denoiser")
+        modelDir = impRes.files("graphite") / "pretrained_models" / "denoiser"
         if self.structure == "SiO2":
             return modelDir.joinpath("SiO2-denoiser.pt")
         elif (