Update README.md

forgot an import in the example