fixed unit test to match extended functionality