small modification on T5