Minor improvement to normal equation