PE-1648 - Fix unit test.