Fixing context bug.