diff --git a/lib/Doctrine/ORM/Mapping/Driver/DriverChain.php b/lib/Doctrine/ORM/Mapping/Driver/DriverChain.php
index 267e9d534..bdeb97b9d 100644
--- a/lib/Doctrine/ORM/Mapping/Driver/DriverChain.php
+++ b/lib/Doctrine/ORM/Mapping/Driver/DriverChain.php
@@ -61,9 +61,9 @@ class DriverChain implements MappingDriver
     /**
      * Set the default driver.
      *
-     * @param Driver $driver
+     * @param MappingDriver $driver
      */
-    public function setDefaultDriver(Driver $driver)
+    public function setDefaultDriver(MappingDriver $driver)
     {
         $this->defaultDriver = $driver;
     }
diff --git a/tests/Doctrine/Tests/ORM/Mapping/DriverChainTest.php b/tests/Doctrine/Tests/ORM/Mapping/DriverChainTest.php
index f5fd60a60..76ca9d29f 100644
--- a/tests/Doctrine/Tests/ORM/Mapping/DriverChainTest.php
+++ b/tests/Doctrine/Tests/ORM/Mapping/DriverChainTest.php
@@ -93,8 +93,8 @@ class DriverChainTest extends \Doctrine\Tests\OrmTestCase
      */
     public function testDefaultDriver()
     {
-        $companyDriver      = $this->getMock('Doctrine\ORM\Mapping\Driver\Driver');
-        $dafaultDriver      = $this->getMock('Doctrine\ORM\Mapping\Driver\Driver');
+        $companyDriver      = $this->getMock('Doctrine\Common\Persistence\Mapping\Driver\MappingDriver');
+        $dafaultDriver      = $this->getMock('Doctrine\Common\Persistence\Mapping\Driver\MappingDriver');
         $entityClassName    = 'Doctrine\Tests\ORM\Mapping\DriverChainEntity';
         $managerClassName   = 'Doctrine\Tests\Models\Company\CompanyManager';
         $chain              = new DriverChain();