Check your VSS/input shaft values. If your cluster is correct, but the Vehicle Speed (SAE) PID is wrong, chances are the improper speed is being reported to the PCM, screwing the shifts.
Good place to start would be crafting a config file in HP Tuners with trans related items and getting a few scans.