Unfortunately the app doesn't restore the state after a configuration change for a bug in the navigation library. I filed an issue in the tracker https://issuetracker.google.com/issues/174158395