diff --git a/stdlib/sql_test.go b/stdlib/sql_test.go
index bfeb07c6..1aa1f261 100644
--- a/stdlib/sql_test.go
+++ b/stdlib/sql_test.go
@@ -781,8 +781,8 @@ func TestBeginTxContextCancelWithDeadConn(t *testing.T) {
 	cancelFn()
 
 	err = tx.Commit()
-	if err != context.Canceled {
-		t.Fatalf("err => %v, want %v", err, context.Canceled)
+	if err != context.Canceled && err != sql.ErrTxDone {
+		t.Fatalf("err => %v, want %v or %v", err, context.Canceled, sql.ErrTxDone)
 	}
 
 	if err := <-errChan; err != nil {