"It seems Khloé needs a good distraction to fully move on mentally from Tristan," a source tells PEOPLE