Fix back button handling when editing comment