[FIX] Do a squared mag instead of mag