diff --git a/src/ucp/proto/proto_init.c b/src/ucp/proto/proto_init.c
index c3da5194ec3..452fcea2347 100644
--- a/src/ucp/proto/proto_init.c
+++ b/src/ucp/proto/proto_init.c
@@ -497,8 +497,8 @@ ucp_proto_common_init_send_perf(const ucp_proto_common_init_params_t *params,
         ucs_assert(reg_md_map == 0);
 
         if ((params->flags & UCP_PROTO_COMMON_INIT_FLAG_REMOTE_ACCESS) &&
-            (params->super.rkey_config_key != NULL)) {
-            remote_mem_type = params->super.rkey_config_key->mem_type;
+            (params->reg_mem_type != UCS_MEMORY_TYPE_UNKNOWN)) {
+            remote_mem_type = params->reg_mem_type;
         } else {
             remote_mem_type = UCS_MEMORY_TYPE_HOST;
         }