diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py
index 02195648dabd9f65197c0f124414484925dfc0ad..d98a41eb203a3646ae4355e5af249268bb883ec9 100644
--- a/model_server/extensions/ilastik/tests/test_ilastik.py
+++ b/model_server/extensions/ilastik/tests/test_ilastik.py
@@ -344,6 +344,15 @@ class TestIlastikOnMultichannelInputs(TestServerBaseClass):
         mod = ilm.IlastikPixelClassifierModel(params={'project_file': self.pa_px_classifier})
         pxmap = mod.infer(self.input_image)[0]
         self.assertEqual(pxmap.hw, self.input_image.hw)
+        self.assertEqual(pxmap.nz, self.input_image.nz)
+        return pxmap
+
+    def test_classify_objects(self):
+        pxmap = self.test_classify_pixels()
+        mod = ilm.IlastikObjectClassifierFromPixelPredictionsModel(params={'project_file': self.pa_ob_classifier})
+        obmap = mod.infer(self.input_image, pxmap)[0]
+        self.assertEqual(obmap.hw, self.input_image.hw)
+        self.assertEqual(obmap.nz, self.input_image.nz)