extended the unit test a bit