improved the unit tests