Add missing ammo check back