Fix missing import in example