Fix crash when on distributed memory machines.

Oops! Missed this one.