Fix missing context