Fix import error in example