Fix where the context failed to get passed.