Fix small bug in melee checking