fixing the unit tests