Fix module import in test_phase_correlate.py