Slight tweak to fix walking and attack