diff --git a/src/webapp/databricks.py b/src/webapp/databricks.py
index 0b57421f..dc22e933 100644
--- a/src/webapp/databricks.py
+++ b/src/webapp/databricks.py
@@ -8,6 +8,7 @@
     Format,
     ExecuteStatementRequestOnWaitTimeout,
     Disposition,
+    StatementState,
 )
 from .config import databricks_vars, gcs_vars
 from .utilities import databricksify_inst_name, SchemaType
@@ -241,7 +242,7 @@ def fetch_table_data(
             raise ValueError(f"Databricks API call failed: {e}")
 
         # Check if the query execution was successful
-        if response.status.state != "SUCCEEDED":
+        if response.status.state != StatementState.SUCCEEDED:
             error_message = (
                 response.status.error.message
                 if response.status.error