Add `with context` to import