PS-239: Fix broken unit test