Fix test

NeurodataWithoutBorders · Jun 18, 2021 · d8ed9fe · d8ed9fe
1 parent 63e6bb4
commit d8ed9fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/test_ophys.py b/tests/unit/test_ophys.py
@@ -234,7 +234,7 @@ def test_init(self):
         )
 
         motion_correction = MotionCorrection(corrected_image_stacks=[corrected_image_stack])
-        self.assertEqual(motion_correction.corrected_image_stacks[0], corrected_image_stack)
+        self.assertEqual(motion_correction.corrected_image_stacks['CorrectedImageStack'], corrected_image_stack)
 
 
 class CorrectedImageStackConstructor(TestCase):