fix build break
2 files changed