Fix test WorkDoneOnEntitiesTest