DOC: fix an import in the example code in README.md.