diff --git a/CHANGELOG.md b/CHANGELOG.md
index 93c813e..2c6f164 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,11 @@
+# Version 2.0.2
+
+* Fixed the connection proxy class created by the factory
+
+# Version 2.0.1
+
+* Fixed next execution time not increasing for scheduled dataflows
+
 # Version 2.0.0
 
 * Add Doctrine DBAL multi-connection support
diff --git a/src/Resources/config/services.yaml b/src/Resources/config/services.yaml
index 52ddc42..a6ccbc4 100644
--- a/src/Resources/config/services.yaml
+++ b/src/Resources/config/services.yaml
@@ -61,7 +61,7 @@ services:
 
     coderhapsodie.dataflow.connection.internal:
         lazy: true
-        class: Doctrine\DBAL\Driver\Connection
+        class: Doctrine\DBAL\Connection
         factory: ['@CodeRhapsodie\DataflowBundle\Factory\ConnectionFactory', 'getConnection']
 
     CodeRhapsodie\DataflowBundle\Factory\ConnectionFactory: