Simplify fftshift test slightly