fix build so all tests pass