Thanks, I've filed an issue: https://github.com/google/mesop/issues/359