diff --git a/app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt b/app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt
index 7556213d5..b7f6f633e 100644
--- a/app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt
+++ b/app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt
@@ -85,8 +85,8 @@ fun NiaApp(
     ),
 ) {
     val background: @Composable (@Composable () -> Unit) -> Unit =
-        when (appState.currentDestination?.route) {
-            TopLevelDestination.FOR_YOU.name -> {
+        when (appState.currentTopLevelDestination) {
+            TopLevelDestination.FOR_YOU -> {
                 content ->
                 NiaGradientBackground(content = content)
             }