add missing changes to fix build