fix build: added missing changes