fix build issue
2 files changed