Fix FlxMath import