diff --git a/query/src/main/java/tech/ydb/query/impl/SessionImpl.java b/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
index b6a03388..b533b2fd 100644
--- a/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
+++ b/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
@@ -315,6 +315,7 @@ public CompletableFuture<Result<QueryInfo>> execute(PartsHandler handler) {
                     result.completeExceptionally(th);
                 }
                 if (status != null) {
+                    updateSessionState(status);
                     if (status.isSuccess()) {
                         result.complete(Result.success(new QueryInfo(stats.get()), status));
                     } else {