Fix sqrt root

Thanks Liyuan for pointing it out.