It's a bit unfortunate that we have to pass this to the block manager through a system property. Can you add it to the constructor of BlockManager / ShuffleBlocksPool instead? I believe that gets created in SparkEnv so you could pass it to SparkEnv.