Weapon fix
Fixed weapon check