Fixed test_imaging.py for 3D