diff --git a/src/QuerySpanTrait.php b/src/QuerySpanTrait.php
index be6b012..e414866 100644
--- a/src/QuerySpanTrait.php
+++ b/src/QuerySpanTrait.php
@@ -39,12 +39,14 @@ public function addTransactionSpan(LoggedQuery $query, ?string $connectionName =
             return;
         }
 
-        if ($context['query'] === 'COMMIT') {
+        if (in_array($context['query'], ['COMMIT', 'ROLLBACK'], true)) {
             $span = $this->popSpan();
 
             if ($span !== null) {
                 $span->finish();
-                $span->setStatus(SpanStatus::ok());
+                if ($context['query'] === 'COMMIT') {
+                    $span->setStatus(SpanStatus::ok());
+                }
             }
 
             return;