Fixed the drive train command

there was left over code from the revert fixed it