diff --git a/lib/internal/Magento/Framework/ObjectManager/Test/Unit/DefinitionFactoryTest.php b/lib/internal/Magento/Framework/ObjectManager/Test/Unit/DefinitionFactoryTest.php
index ca086c23aefe4db84f89ec38311a2c5b602a3b17..8ad1ac13a12cdac1c8ea8ed1d993c0bb258f0809 100644
--- a/lib/internal/Magento/Framework/ObjectManager/Test/Unit/DefinitionFactoryTest.php
+++ b/lib/internal/Magento/Framework/ObjectManager/Test/Unit/DefinitionFactoryTest.php
@@ -7,14 +7,14 @@ namespace Magento\Framework\ObjectManager\Test\Unit;
 
 use Magento\Framework\ObjectManager\Definition\Compiled;
 use Magento\Framework\Serialize\SerializerInterface;
-use Magento\Framework\Filesystem\DriverInterface;
+use Magento\Framework\Filesystem\Driver\File;
 use Magento\Framework\ObjectManager\DefinitionFactory;
 use Magento\Framework\ObjectManager\Definition\Runtime;
 
 class DefinitionFactoryTest extends \PHPUnit_Framework_TestCase
 {
     /**
-     * @var DriverInterface|\PHPUnit_Framework_MockObject_MockObject
+     * @var File|\PHPUnit_Framework_MockObject_MockObject
      */
     private $filesystemDriverMock;
 
@@ -30,7 +30,7 @@ class DefinitionFactoryTest extends \PHPUnit_Framework_TestCase
 
     protected function setUp()
     {
-        $this->filesystemDriverMock = $this->getMock(DriverInterface::class);
+        $this->filesystemDriverMock = $this->getMock(File::class);
         $this->serializerMock = $this->getMock(SerializerInterface::class);
         $this->definitionFactory = new DefinitionFactory(
             $this->filesystemDriverMock,