diff --git a/test/test_learn_wave_quadratic_phase_function.py b/test/test_learn_wave_quadratic_phase_function.py
index 9cfdd72d..e01a05ae 100644
--- a/test/test_learn_wave_quadratic_phase_function.py
+++ b/test/test_learn_wave_quadratic_phase_function.py
@@ -58,7 +58,6 @@ def test(
                                             device = device
                                            )
     reconstruction_intensities = propagator.reconstruct(hologram_phases, amplitude = None)
-    import sys; sys.exit()
     for frame_id in range(reconstruction_intensities.shape[0]):
         for depth_id in range(reconstruction_intensities.shape[1]):
             reconstruction_intensity = reconstruction_intensities[frame_id, depth_id]