fixed unit test again