updated the unit tests