add missing import in example code