Fix compile problem
2 files changed