diff --git a/src/Symfony/Component/DependencyInjection/Configuration/Builder/ExprBuilder.php b/src/Symfony/Component/DependencyInjection/Configuration/Builder/ExprBuilder.php
index b0e63cc35e..3c9686448e 100644
--- a/src/Symfony/Component/DependencyInjection/Configuration/Builder/ExprBuilder.php
+++ b/src/Symfony/Component/DependencyInjection/Configuration/Builder/ExprBuilder.php
@@ -101,11 +101,11 @@ class ExprBuilder
      */
     public function thenReplaceKeyWithAttribute($attribute)
     {
-        $this->thenPart = function($v) {
+        $this->thenPart = function($v) use ($attribute) {
             $newValue = array();
             foreach ($v as $k => $oldValue) {
-                if (is_array($oldValue) && isset($oldValue['id'])) {
-                    $k = $oldValue['id'];
+                if (is_array($oldValue) && isset($oldValue[$attribute])) {
+                    $k = $oldValue[$attribute];
                 }
 
                 $newValue[$k] = $oldValue;