Fix unit test #2