diff --git a/dbt_cloud_plugin/dbt_cloud/dbt_cloud.py b/dbt_cloud_plugin/dbt_cloud/dbt_cloud.py
index 7b0afdb..7537b71 100644
--- a/dbt_cloud_plugin/dbt_cloud/dbt_cloud.py
+++ b/dbt_cloud_plugin/dbt_cloud/dbt_cloud.py
@@ -3,6 +3,8 @@
 import requests
 import time
 
+from airflow.exceptions import AirflowException
+
 class DbtCloud(object):
     """
     Class for interacting with the dbt Cloud API