Fix unit test example.