Fix null context error