diff --git a/xla/python/ifrt_proxy/integration_tests/mock_array_test.cc b/xla/python/ifrt_proxy/integration_tests/mock_array_test.cc
index 62f1e7cad1601..c58b626e71e8c 100644
--- a/xla/python/ifrt_proxy/integration_tests/mock_array_test.cc
+++ b/xla/python/ifrt_proxy/integration_tests/mock_array_test.cc
@@ -116,6 +116,7 @@ class ProxyWithMockBackend {
         client_->MakeArrayFromHostBuffer(
             data->data(), dtype, shape,
             /*byte_strides=*/std::nullopt, sharding,
+            /*layout=*/nullptr,
             Client::HostBufferSemantics::kImmutableOnlyDuringCall,
             /*on_done_with_host_buffer=*/nullptr));