just compute norm to avoid sqrt here