Fix weapon cooldown. Weapon now moves with the unit again