From af4ea3d87a25c87abe076e20f84cebaaafe4b93f Mon Sep 17 00:00:00 2001
From: Alessandro Lollo <alessandro.lollo@cloudacademy.com>
Date: Wed, 20 Apr 2022 22:17:22 +0200
Subject: [PATCH] Fixed docstring

---
 prefect_cubejs/tasks.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/prefect_cubejs/tasks.py b/prefect_cubejs/tasks.py
index 19cbbd4..c5b1c93 100644
--- a/prefect_cubejs/tasks.py
+++ b/prefect_cubejs/tasks.py
@@ -68,8 +68,8 @@ def run_query(
         - `CubeJSConfigurationException` if `api_token` is missing
             and `api_token_env_var` cannot be found.
         - `CubeJSConfigurationException` if `query` is missing.
-        - `prefect.engine.signals.FAIL` if the Cube.js load API fails.
-        - `prefect.engine.signals.FAIL` if the Cube.js load API takes more than
+        - `CubeJSAPIFailureException` if the Cube.js load API fails.
+        - `CubeJSAPIFailureException` if the Cube.js load API takes more than
             `max_wait_time` seconds to respond.
 
     Returns: