Fix unit test logic