diff --git a/run_alphafold_data_test.py b/run_alphafold_data_test.py
index 74ddbed..0e15afd 100644
--- a/run_alphafold_data_test.py
+++ b/run_alphafold_data_test.py
@@ -31,7 +31,6 @@ from alphafold3.constants import chemical_components
 from alphafold3.data import featurisation
 from alphafold3.data import pipeline
 from alphafold3.model.atom_layout import atom_layout
-from alphafold3.structure import test_utils
 import jax
 import numpy as np
 
@@ -98,7 +97,7 @@ def _generate_diff(actual: str, expected: str) -> str:
   )
 
 
-class DataPipelineTest(test_utils.StructureTestCase):
+class DataPipelineTest(parameterized.TestCase):
   """Test AlphaFold 3 inference."""
 
   def setUp(self):
diff --git a/run_alphafold_test.py b/run_alphafold_test.py
index 22d0787..bd984bf 100644
--- a/run_alphafold_test.py
+++ b/run_alphafold_test.py
@@ -28,7 +28,6 @@ from alphafold3.common import resources
 from alphafold3.common.testing import data as testing_data
 from alphafold3.data import pipeline
 from alphafold3.model.scoring import alignment
-from alphafold3.structure import test_utils
 import jax
 import numpy as np
 
@@ -64,7 +63,7 @@ def _generate_diff(actual: str, expected: str) -> str:
   )
 
 
-class InferenceTest(test_utils.StructureTestCase):
+class InferenceTest(parameterized.TestCase):
   """Test AlphaFold 3 inference."""
 
   def setUp(self):