small changes to fix build error