return context error if the context was canceled mid-way

return context error if the context was canceled mid-way #1056