diff --git a/goodman_pipeline/core/tests/test_core.py b/goodman_pipeline/core/tests/test_core.py
index 4229a2dc..53636df5 100644
--- a/goodman_pipeline/core/tests/test_core.py
+++ b/goodman_pipeline/core/tests/test_core.py
@@ -922,7 +922,7 @@ def test_extraction_moffat(self):
         np.testing.assert_array_almost_equal(extracted, self.reference_result_moffat)
 
     def test_extraction_not_implemented_model(self):
-        spatial_profile = models.BlackBody1D()
+        spatial_profile = models.BlackBody()
         self.assertRaises(NotImplementedError, extraction, ccd=self.fake_image,
                           target_trace=self.target_trace,
                           spatial_profile=spatial_profile,
@@ -2281,7 +2281,7 @@ def test_trace_not_implemented(self):
         self.assertRaises(NotImplementedError,
                           trace,
                           self.ccd2,
-                          models.BlackBody1D(),
+                          models.BlackBody(),
                           trace_model,
                           fitter,
                           5)