Update sq_norm as well