diff --git a/src/Symfony/Component/Form/Tests/Extension/Validator/Constraints/FormValidatorTest.php b/src/Symfony/Component/Form/Tests/Extension/Validator/Constraints/FormValidatorTest.php
index 48f6123426..f1ee046551 100644
--- a/src/Symfony/Component/Form/Tests/Extension/Validator/Constraints/FormValidatorTest.php
+++ b/src/Symfony/Component/Form/Tests/Extension/Validator/Constraints/FormValidatorTest.php
@@ -293,17 +293,16 @@ class FormValidatorTest extends \PHPUnit_Framework_TestCase
 
     public function testDontExecuteFunctionNames()
     {
-        $context = $this->getExecutionContext();
-        $graphWalker = $context->getGraphWalker();
+        $context = $this->getMockExecutionContext();
         $object = $this->getMock('\stdClass');
         $options = array('validation_groups' => 'header');
         $form = $this->getBuilder('name', '\stdClass', $options)
             ->setData($object)
             ->getForm();
 
-        $graphWalker->expects($this->once())
-            ->method('walkReference')
-            ->with($object, 'header', 'data', true);
+        $context->expects($this->once())
+            ->method('validate')
+            ->with($object, 'data', 'header', true);
 
         $this->validator->initialize($context);
         $this->validator->validate($form, new Form());