Fixed AI checking