updating the unit tests