fix build minor bug
fix build