Move sqrt import

Since * is imported from mesh, this stomped on UFL's sqrt.