fix role predict of AI
Revert "fix role predict of AI"

This reverts commit b38d9d78c3f4b4247a982a2c8ba0428ca3bfaa7d.
another temp way to fix role predict in AI