fix done not logging when using with_context in a worker