Added missing 'done' callback