diff --git a/feature/foryou/src/androidTest/java/com/google/samples/apps/nowinandroid/feature/foryou/ForYouRobot.kt b/feature/foryou/src/androidTest/java/com/google/samples/apps/nowinandroid/feature/foryou/ForYouRobot.kt
index 02d113389..79a1888b2 100644
--- a/feature/foryou/src/androidTest/java/com/google/samples/apps/nowinandroid/feature/foryou/ForYouRobot.kt
+++ b/feature/foryou/src/androidTest/java/com/google/samples/apps/nowinandroid/feature/foryou/ForYouRobot.kt
@@ -45,14 +45,14 @@ internal class ForYouRobot(
 
     fun setContent(
         isSyncing: Boolean,
-        interestsSelectionState: ForYouInterestsSelectionUiState,
+        onboardingState: OnboardingUiState,
         feedState: NewsFeedUiState,
     ) {
         composeTestRule.setContent {
             BoxWithConstraints {
                 ForYouScreen(
                     isSyncing = isSyncing,
-                    interestsSelectionState = interestsSelectionState,
+                    onboardingUiState = onboardingState,
                     feedState = feedState,
                     onTopicCheckedChanged = { _, _ -> },
                     onAuthorCheckedChanged = { _, _ -> },