fixed unit test that was broekn