Fix unit test that relies on version