Fix my typo from an earlier change.
Fix my typo from an earlier change.

PiperOrigin-RevId: 226033790

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