diff --git a/tests/order/test-wc-retailcrm-order-address.php b/tests/order/test-wc-retailcrm-order-address.php
index b7c0a52..945b8d9 100644
--- a/tests/order/test-wc-retailcrm-order-address.php
+++ b/tests/order/test-wc-retailcrm-order-address.php
@@ -26,8 +26,11 @@ class WC_Retailcrm_Order_Address_Test extends WC_Retailcrm_Test_Case_Helper
 
     public function test_build()
     {
-        $order_address = new WC_Retailcrm_Order_Address;
-        $data = $order_address->build($this->order)->get_data();
+        $order_address = new WC_Retailcrm_Order_Address();
+        $data = $order_address
+	        ->setWCAddressType(WC_Retailcrm_Abstracts_Address::ADDRESS_TYPE_BILLING)
+	        ->build($this->order)
+	        ->get_data();
 
         $this->assertArrayHasKey('index', $data);
         $this->assertArrayHasKey('city', $data);