Fixed issue with AI stuck in melee attack

Add the highlighted nodes shown in the image, location at the top of the screenshot.