Fixed import in example