Fix a bug in ai-test suite.