OZP-64: Fixing unit test