[FIX] Error when context is None