We actually added a parameter for this in the code on branch-0.6, called spark.akka.frameSize. Maybe you should copy in the version of the function from there to stay consistent.