diff --git a/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php b/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php
index bc6387f0a2683e2b7f0c4d5d83115953355d21e7..0c23c1c21a10d8aac8e779b041b5a52c00f10303 100644
--- a/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php
+++ b/dev/tests/unit/testsuite/Magento/Setup/Controller/ModulesTest.php
@@ -34,12 +34,15 @@ class ModulesTest extends \PHPUnit_Framework_TestCase
     {
         $this->objectManager = $this->getMockForAbstractClass('Magento\Framework\ObjectManagerInterface');
         $this->status = $this->getMock('Magento\Framework\Module\Status', [], [], '', false);
-        $objectManagerFactory = $this->getMock('Magento\Setup\Model\ObjectManagerFactory', [], [], '', false);
-        $objectManagerFactory->expects($this->once())->method('create')->willReturn($this->objectManager);
+        /** @var
+         * $objectManagerProvider \PHPUnit_Framework_MockObject_MockObject|\Magento\Setup\Model\ObjectManagerProvider
+         */
+        $objectManagerProvider = $this->getMock('Magento\Setup\Model\ObjectManagerProvider', [], [], '', false);
+        $objectManagerProvider->expects($this->once())->method('get')->willReturn($this->objectManager);
         $this->modules = $this->getMock('Magento\Setup\Model\ModuleStatus', [], [], '', false);
         $this->status = $this->getMock('Magento\Framework\Module\Status', [], [], '', false);
         $this->objectManager->expects($this->once())->method('create')->will($this->returnValue($this->status));
-        $this->controller = new Modules($this->modules, $objectManagerFactory);
+        $this->controller = new Modules($this->modules, $objectManagerProvider);
     }
 
     /**