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 502dc43ff316d6776ce6a4c00483371ff3a546a0..3587327697a3320030f043c964e1b3b857b7ea03 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
@@ -62,6 +62,7 @@ class DownloadableTest extends \PHPUnit_Framework_TestCase
         $this->extensionAttributesMock = $this->getMockBuilder(
                 \Magento\Catalog\Api\Data\ProductExtensionInterface::class
             )->disableOriginalConstructor()
+            ->setMethods(['setDownloadableProductSamples', 'setDownloadableProductLinks'])
             ->getMockForAbstractClass();
         $this->sampleFactoryMock = $this->getMockBuilder('\Magento\Downloadable\Api\Data\SampleInterfaceFactory')
             ->disableOriginalConstructor()