diff --git a/dbt_project.yml b/dbt_project.yml
index 635a4a8..28fd4bf 100644
--- a/dbt_project.yml
+++ b/dbt_project.yml
@@ -60,3 +60,14 @@ vars:
   STREAMLINE_USE_DEV_FOR_EXTERNAL_TABLES: False
   STREAMLINE_INVOKE_STREAMS: False
   STREAMLINE_RUN_HISTORY: False
+
+dispatch:
+  - macro_namespace: dbt
+    search_order:
+      - flow-models
+      - dbt_snowflake_query_tags
+      - dbt
+
+query-comment:
+  comment: '{{ dbt_snowflake_query_tags.get_query_comment(node) }}'
+  append: true # Snowflake removes prefixed comments.
diff --git a/macros/custom_query_tag.sql b/macros/custom_query_tag.sql
deleted file mode 100644
index 809e1bb..0000000
--- a/macros/custom_query_tag.sql
+++ /dev/null
@@ -1,11 +0,0 @@
-{% macro set_query_tag() -%}
- {% set new_json = {"repo":project_name, "object":this.table, "profile":target.profile_name, "env":target.name, "existing_tag":get_current_query_tag()  } %}
-{% set new_query_tag = tojson(new_json) | as_text %}
-  {% if new_query_tag %}
-    {% set original_query_tag = get_current_query_tag() %}
-    {{ log("Setting query_tag to '" ~ new_query_tag ~ "'. Will reset to '" ~ original_query_tag ~ "' after materialization.") }}
-    {% do run_query("alter session set query_tag = '{}'".format(new_query_tag)) %}
-    {{ return(original_query_tag)}}
-  {% endif %}
-  {{ return(none)}}
-{% endmacro %}
\ No newline at end of file
diff --git a/packages.yml b/packages.yml
index 6e26f32..feccffa 100644
--- a/packages.yml
+++ b/packages.yml
@@ -7,3 +7,5 @@ packages:
     version: 1.0.0
   - git: https://github.com/FlipsideCrypto/fsc-utils.git
     revision: "v1.3.0"
+  - package: get-select/dbt_snowflake_query_tags
+    version: [">=2.0.0", "<3.0.0"]