AI bug fixes