Fixed import in README example