diff --git a/app/code/Magento/Quote/Model/PaymentMethodManagement.php b/app/code/Magento/Quote/Model/PaymentMethodManagement.php
index 925a4e2ab37d33d79a6470701bafdd38367ce353..d29e3732d747f9b920c40ee2931040b8b70a8907 100644
--- a/app/code/Magento/Quote/Model/PaymentMethodManagement.php
+++ b/app/code/Magento/Quote/Model/PaymentMethodManagement.php
@@ -27,11 +27,6 @@ class PaymentMethodManagement implements \Magento\Quote\Api\PaymentMethodManagem
      */
     protected $methodList;
 
-    /**
-     * @var
-     */
-    protected $addressFactory;
-
     /**
      * Constructor
      *
@@ -74,10 +69,6 @@ class PaymentMethodManagement implements \Magento\Quote\Api\PaymentMethodManagem
         $payment->importData($data);
 
         if ($quote->isVirtual()) {
-            // check if billing address is set
-//            if ($quote->getBillingAddress()->getCountryId() === null) {
-//                throw new InvalidTransitionException(__('Billing address is not set'));
-//            }
             $quote->getBillingAddress()->setPaymentMethod($payment->getMethod());
         } else {
             // check if shipping address is set