make the unit tests pass too