diff --git a/test/integration/tst_regression.py b/test/integration/tst_regression.py
index <HASH>..<HASH> 100644
--- a/test/integration/tst_regression.py
+++ b/test/integration/tst_regression.py
@@ -2,8 +2,7 @@ import shutil
 import os
 
 import autofit as af
-from autolens.data.instrument import ccd
-from autolens.data import ccd as ccd
+from autolens.data.instrument import ccd as ccd
 from autolens.data.array import grids
 from autolens.data.array.util import array_util
 from autolens.lens import ray_tracing
diff --git a/test/simulation/makers.py b/test/simulation/makers.py
index <HASH>..<HASH> 100644
--- a/test/simulation/makers.py
+++ b/test/simulation/makers.py
@@ -1,6 +1,5 @@
 import autofit as af
 from autolens.data.instrument import ccd
-from autolens.data import ccd
 from autolens.data.array import grids
 from autolens.lens import ray_tracing
 from autolens.model.galaxy import galaxy as g
Refactored data to use instrument module with abstract_data,
ready to add interferometry data class.