Fix overflow in test_sq