diff --git a/app/code/Magento/Persistent/Test/Unit/Observer/EmulateCustomerObserverTest.php b/app/code/Magento/Persistent/Test/Unit/Observer/EmulateCustomerObserverTest.php
index 6c35ade65451b..2df36577b2931 100644
--- a/app/code/Magento/Persistent/Test/Unit/Observer/EmulateCustomerObserverTest.php
+++ b/app/code/Magento/Persistent/Test/Unit/Observer/EmulateCustomerObserverTest.php
@@ -131,14 +131,14 @@ public function testExecuteWhenSessionPersistAndCustomerNotLoggedIn()
         $customerMock
             ->expects($this->once())
             ->method('getDefaultShipping')
-            ->willReturn('shippingId');
+            ->willReturn(12345);
         $customerMock
             ->expects($this->once())
             ->method('getDefaultBilling')
-            ->willReturn('billingId');
+            ->willReturn(12346);
         $valueMap = [
-            ['shippingId', $defaultShippingAddressMock],
-            ['billingId', $defaultBillingAddressMock]
+            [12345, $defaultShippingAddressMock],
+            [12346, $defaultBillingAddressMock]
         ];
         $this->addressRepositoryMock->expects($this->any())->method('getById')->willReturnMap($valueMap);
         $this->customerSessionMock