fix build errors
Fix build but!