fix build errors: missing import