Fix for the weapon check