diff --git a/descarteslabs/core/catalog/tests/test_image.py b/descarteslabs/core/catalog/tests/test_image.py
index 1f83190f..485d6186 100644
--- a/descarteslabs/core/catalog/tests/test_image.py
+++ b/descarteslabs/core/catalog/tests/test_image.py
@@ -983,7 +983,7 @@ def test_ndarray_geocontext(self, mock_raster):
         image = Image.get("landsat:LC08:PRE:TOAR:meta_LC80270312016188_v1")
         arr, info = image.ndarray("red", resolution=1000, raster_info=True)
 
-        assert mock_raster.called_once()
+        mock_raster.assert_called_once()
         assert image.geocontext.geometry is not None
         print(mock_raster.call_args)
         assert mock_raster.call_args[1]["cutline"] is None