Fix issue with sqrt