Add CollectUntilIdle