Fixing the AI :(