Fix missing import in README.md code snippet