Fix missing sqrt import