Add back missing context variable.