Make TE edit back button work