diff --git a/app/code/Magento/CheckoutAgreements/Model/Checkout/Plugin/Validation.php b/app/code/Magento/CheckoutAgreements/Model/Checkout/Plugin/Validation.php
index daab7a643e9f3d0805fd3ef08cd3f3dd0f6d1e07..e1c619332339fd1ca64bfa7e1db531fbe751307e 100644
--- a/app/code/Magento/CheckoutAgreements/Model/Checkout/Plugin/Validation.php
+++ b/app/code/Magento/CheckoutAgreements/Model/Checkout/Plugin/Validation.php
@@ -24,18 +24,18 @@ class Validation
     protected $checkoutAgreementsRepository;
 
     /**
-     * @var \Magento\CheckoutAgreements\Model\AgreementsValidator
+     * @var \Magento\Checkout\Api\AgreementsValidatorInterface
      */
     protected $agreementsValidator;
 
     /**
-     * @param \Magento\CheckoutAgreements\Model\AgreementsValidator $agreementsValidator
+     * @param \Magento\Checkout\Api\AgreementsValidatorInterface $agreementsValidator
      * @param \Magento\Framework\App\Config\ScopeConfigInterface $scopeConfiguration
      * @param \Magento\CheckoutAgreements\Api\CheckoutAgreementsRepositoryInterface $checkoutAgreementsRepository
      * @codeCoverageIgnore
      */
     public function __construct(
-        \Magento\CheckoutAgreements\Model\AgreementsValidator $agreementsValidator,
+        \Magento\Checkout\Api\AgreementsValidatorInterface $agreementsValidator,
         \Magento\Framework\App\Config\ScopeConfigInterface $scopeConfiguration,
         \Magento\CheckoutAgreements\Api\CheckoutAgreementsRepositoryInterface $checkoutAgreementsRepository
     ) {
diff --git a/app/code/Magento/CheckoutAgreements/Test/Unit/Model/Checkout/Plugin/ValidationTest.php b/app/code/Magento/CheckoutAgreements/Test/Unit/Model/Checkout/Plugin/ValidationTest.php
index 6889aa987d9e4ff18cc107325ddc9e5c21af604a..beab6679ed594ec2c7d8eb2c111ac6b9adc58f01 100644
--- a/app/code/Magento/CheckoutAgreements/Test/Unit/Model/Checkout/Plugin/ValidationTest.php
+++ b/app/code/Magento/CheckoutAgreements/Test/Unit/Model/Checkout/Plugin/ValidationTest.php
@@ -52,13 +52,7 @@ class ValidationTest extends \PHPUnit_Framework_TestCase
 
     protected function setUp()
     {
-        $this->agreementsValidatorMock = $this->getMock(
-            '\Magento\CheckoutAgreements\Model\AgreementsValidator',
-            [],
-            [],
-            '',
-            false
-        );
+        $this->agreementsValidatorMock = $this->getMock('\Magento\Checkout\Api\AgreementsValidatorInterface');
         $this->subjectMock = $this->getMock('\Magento\Checkout\Api\PaymentInformationManagementInterface');
         $this->paymentMock = $this->getMock('\Magento\Quote\Api\Data\PaymentInterface');
         $this->addressMock = $this->getMock('\Magento\Quote\Api\Data\AddressInterface');