diff --git a/piperider_cli/dbt/list_task.py b/piperider_cli/dbt/list_task.py
index 8d959f028..24f6c6a7b 100644
--- a/piperider_cli/dbt/list_task.py
+++ b/piperider_cli/dbt/list_task.py
@@ -9,9 +9,9 @@
 import agate
 import dbt.flags as flags_module
 from dbt.adapters.base import BaseAdapter, BaseRelation, Column as BaseColumn
+from dbt.adapters.contracts.connection import QueryComment
 from dbt.config.project import VarProvider
 from dbt.config.runtime import RuntimeConfig
-from dbt.contracts.connection import QueryComment
 from dbt.contracts.graph.manifest import Manifest, WritableManifest
 from dbt.contracts.project import PackageConfig, UserConfig
 from dbt.contracts.state import PreviousState