fixed compile error
Merge https://github.com/qqliu/IDIOT