correct all the unit tests