Fix unit test to pass