diff --git a/superset/db_engine_specs.py b/superset/db_engine_specs.py
index c2ade542e0f99..75db166cb8ce6 100644
--- a/superset/db_engine_specs.py
+++ b/superset/db_engine_specs.py
@@ -711,7 +711,7 @@ def handle_cursor(cls, cursor, query, session):
         )
         polled = cursor.poll()
         while polled.operationState in unfinished_states:
-            query = session.query(type(query)).filter_by(id=query.id)
+            query = session.query(type(query)).filter_by(id=query.id).one()
             if query.status == QueryStatus.STOPPED:
                 cursor.cancel()
                 break