Change comparing real and imaginary with zero