Small fix for returning from EditFragment