diff --git a/tests/test_displacements.py b/tests/test_displacements.py
index 76d2a054..9f1acbe0 100644
--- a/tests/test_displacements.py
+++ b/tests/test_displacements.py
@@ -210,14 +210,14 @@ def test_plot_site_displacements_ytos(self):
     
     @custom_mpl_image_compare(filename="v_Cd_0_disp_ellipsoid_plot.png")
     def test_calc_displacements_ellipsoid_ellipsoid(self):
-        defect_entry = core.DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
+        defect_entry = DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
         return_tuple = calc_displacements_ellipsoid(defect_entry, plot_ellipsoid=True)
         # return fig object
         return return_tuple[3]
     
     @custom_mpl_image_compare(filename="v_Cd_0_disp_anisotropy_plot.png")
     def test_calc_displacements_ellipsoid_anisotropy(self):
-        defect_entry = core.DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
+        defect_entry = DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
         return_tuple = calc_displacements_ellipsoid(defect_entry, plot_anisotropy=True)
         # return fig object
         return return_tuple[3]