Fix broken build, truly :)
Fix broken build, truly :)

PiperOrigin-RevId: 222286456

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor