Fixes AI behavior