Tried to fix AI bug...