fix unit test

fixed unit test that was no longer valid