Fixed AI freeze