Fix enemy AI