Was the work performed by Kenworth or Allison Transmission?
Not really being totally familiar with your transmission or smaller ones I 'm thinking or wondering to myself if they cleaned the oil cooler and lines after they did the other repairs.
How old or how many miles are on the unit?