In the wake of Tristan's cheating scandal, a source tells ET that Khloe is now planning on returning to California as soon as the baby is cleared for travel.