Corrected a stupid mistake in AI