Possible fix for imaginary numbers