Add explicit done call to avoid timeout