diff --git a/data/crop_lammps_outputs.py b/data/crop_lammps_outputs.py
index fee07fbf..79a75f78 100644
--- a/data/crop_lammps_outputs.py
+++ b/data/crop_lammps_outputs.py
@@ -5,8 +5,10 @@
 
 import yaml
 
-from crystal_diffusion.data.utils import crop_lammps_yaml
-from crystal_diffusion.utils.logging_utils import setup_analysis_logger
+from diffusion_for_multi_scale_molecular_dynamics.data.utils import \
+    crop_lammps_yaml
+from diffusion_for_multi_scale_molecular_dynamics.utils.logging_utils import \
+    setup_analysis_logger
 
 logger = logging.getLogger(__name__)