Add done channel to stop short.