Fix import in example