From 72c3f65dae8149a0d50a2f8f9efd9752bf9aba1e Mon Sep 17 00:00:00 2001
From: Justin Beaty <justin.beaty@gmail.com>
Date: Thu, 19 Aug 2021 13:50:45 -0700
Subject: [PATCH] Fix address2 / company

---
 app/code/community/Gene/Braintree/Helper/Data.php | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

diff --git a/app/code/community/Gene/Braintree/Helper/Data.php b/app/code/community/Gene/Braintree/Helper/Data.php
index 1144e53..9a9e752 100644
--- a/app/code/community/Gene/Braintree/Helper/Data.php
+++ b/app/code/community/Gene/Braintree/Helper/Data.php
@@ -100,26 +100,18 @@ public function convertToBraintreeAddress($address)
                 'firstName'         => $address->getFirstname(),
                 'lastName'          => $address->getLastname(),
                 'streetAddress'     => $address->getStreet1(),
+                'extendedAddress'   => $address->getStreet2(),
+                'company'           => $address->getCompany(),
                 'locality'          => $address->getCity(),
                 'postalCode'        => $address->getPostcode(),
                 'countryCodeAlpha2' => $address->getCountry()
             );
 
-            // Any extended address?
-            if ($address->getStreet2()) {
-                $return['extendedAddress'] = $address->getStreet2();
-            }
-
             // Region
             if ($address->getRegion()) {
                 $return['region'] = $address->getRegionCode();
             }
 
-            // Check to see if we have a company
-            if ($address->getCompany()) {
-                $return['company'] = $address->getCompany();
-            }
-
             return $return;
         }
     }