Jetpack Compose release version: 1.0.0-beta01

Android Studio Build: #AI-203.7148.57.2031.7185775, built on March 5, 2021

Steps to Reproduce:

Select "Proceed" button (navigating to Second screen).
Press system back button (navigating to Third screen).
Press system back button (navigating back to First screen, without interception).
Lock/unlock screen to restart Activity.
Press system back button (navigating to First screen instead of Third screen - interception stops working).

Code to reproduce the issue (from attached sample project):

class MainActivity : ComponentActivity() {
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContent {
            MaterialTheme {
                Surface(Modifier.fillMaxSize()) {
                    SampleApp()
                }
            }
        }
    }
}

@Composable
fun SampleApp() {
    val navController = rememberNavController()
    val navigateTo: (String) -> Unit = { route -> navController.navigate(route) }
    NavHost(navController, "first") {
        composable("first") { FirstScreen(onClick = { navigateTo("second") }) }
        composable("second") { SecondScreen(onBack = { navigateTo("third") }) }
        composable("third") { ThirdScreen() }
    }
}

@Composable
fun FirstScreen(onClick: () -> Unit) {
    Column {
        Text("First screen")
        Button(onClick) { Text("Proceed") }
    }
}

@Composable
fun SecondScreen(onBack: () -> Unit) {
    BackHandler {
        onBack()
        println("System back intercepted")
    }
    Text("Second screen")
}

@Composable
fun ThirdScreen() {
    Text("Third screen")
}