Fix the bugs that using norm as squared L2