fixes so tests pass