Fix context issue with Done button.