fix sqrt(x**2) != abs(x) bug