diff --git a/app/code/Magento/Downloadable/Test/Unit/Controller/Adminhtml/Product/Initialization/Helper/Plugin/DownloadableTest.php b/app/code/Magento/Downloadable/Test/Unit/Controller/Adminhtml/Product/Initialization/Helper/Plugin/DownloadableTest.php
index dba46e81bcc5b6e3ebce2d02ec095a4e540b4208..502dc43ff316d6776ce6a4c00483371ff3a546a0 100644
--- a/app/code/Magento/Downloadable/Test/Unit/Controller/Adminhtml/Product/Initialization/Helper/Plugin/DownloadableTest.php
+++ b/app/code/Magento/Downloadable/Test/Unit/Controller/Adminhtml/Product/Initialization/Helper/Plugin/DownloadableTest.php
@@ -59,13 +59,10 @@ class DownloadableTest extends \PHPUnit_Framework_TestCase
             '',
             false
         );
-        $this->extensionAttributesMock = $this->getMock(
-            'Magento\Catalog\Api\Data\ProductExtensionInterface',
-            ['setDownloadableProductSamples', 'setDownloadableProductLinks'],
-            [],
-            '',
-            false
-        );
+        $this->extensionAttributesMock = $this->getMockBuilder(
+                \Magento\Catalog\Api\Data\ProductExtensionInterface::class
+            )->disableOriginalConstructor()
+            ->getMockForAbstractClass();
         $this->sampleFactoryMock = $this->getMockBuilder('\Magento\Downloadable\Api\Data\SampleInterfaceFactory')
             ->disableOriginalConstructor()
             ->setMethods(['create'])