OZF-299: Fix broken unit test