Thor headed back once all the eneimies were gone. He knew the suit needed to be fixed for Tony's safety.