diff --git a/composer.json b/composer.json
index 01959bd..e8c009e 100644
--- a/composer.json
+++ b/composer.json
@@ -31,6 +31,9 @@
         "symfony/phpunit-bridge": "^6.4"
     },
     "autoload": {
+        "files": [
+            "src/aliases.php"
+        ],
         "psr-4": {
             "DataDog\\AuditBundle\\": "src"
         }
diff --git a/src/DBAL/Middleware/ConnectionMiddlewareForV3.php b/src/DBAL/Middleware/ConnectionMiddlewareForV3.php
new file mode 100644
index 0000000..a9e1926
--- /dev/null
+++ b/src/DBAL/Middleware/ConnectionMiddlewareForV3.php
@@ -0,0 +1,35 @@
+<?php
+
+namespace DataDog\AuditBundle\DBAL\Middleware;
+
+use Doctrine\DBAL\Driver\Connection;
+use Doctrine\DBAL\Driver\Exception;
+use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
+
+class ConnectionMiddlewareForV3 extends AbstractConnectionMiddleware
+{
+    public function __construct(
+        Connection $wrappedConnection,
+        private readonly AuditFlushMiddleware $auditFlushMiddleware
+    )
+    {
+        parent::__construct($wrappedConnection);
+    }
+
+    /**
+     * Override of the commit method
+     *
+     * @throws Exception
+     */
+    public function commit()
+    {
+        // Call the flusher callback if it's available.
+        if ($this->auditFlushMiddleware->flushHandler !== null) {
+            ($this->auditFlushMiddleware->flushHandler)();
+            $this->auditFlushMiddleware->flushHandler = null;
+        }
+
+        // Call the parent's commit method
+        return parent::commit();
+    }
+}
diff --git a/src/DBAL/Middleware/ConnectionMiddleware.php b/src/DBAL/Middleware/ConnectionMiddlewareForV4.php
similarity index 92%
rename from src/DBAL/Middleware/ConnectionMiddleware.php
rename to src/DBAL/Middleware/ConnectionMiddlewareForV4.php
index fe26309..52c1f4b 100644
--- a/src/DBAL/Middleware/ConnectionMiddleware.php
+++ b/src/DBAL/Middleware/ConnectionMiddlewareForV4.php
@@ -6,7 +6,7 @@
 use Doctrine\DBAL\Driver\Exception;
 use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
 
-class ConnectionMiddleware extends AbstractConnectionMiddleware
+class ConnectionMiddlewareForV4 extends AbstractConnectionMiddleware
 {
     public function __construct(
         Connection $wrappedConnection,
diff --git a/src/aliases.php b/src/aliases.php
new file mode 100644
index 0000000..9ea2cba
--- /dev/null
+++ b/src/aliases.php
@@ -0,0 +1,18 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DataDog\AuditBundle;
+
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddleware;
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddlewareForV3;
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddlewareForV4;
+use Doctrine\DBAL\VersionAwarePlatformDriver;
+
+if (!class_exists(ConnectionMiddleware::class)) {
+    if (!interface_exists(VersionAwarePlatformDriver::class)) {
+        class_alias(ConnectionMiddlewareForV4::class, ConnectionMiddleware::class);
+    } else {
+        class_alias(ConnectionMiddlewareForV3::class, ConnectionMiddleware::class);
+    }
+}