Fixed Bug in AI Coding