diff --git a/app/src/testDemo/kotlin/com/google/samples/apps/nowinandroid/ui/NiaAppScreenSizesScreenshotTests.kt b/app/src/testDemo/kotlin/com/google/samples/apps/nowinandroid/ui/NiaAppScreenSizesScreenshotTests.kt
index 2a2f7bb63..61e88286f 100644
--- a/app/src/testDemo/kotlin/com/google/samples/apps/nowinandroid/ui/NiaAppScreenSizesScreenshotTests.kt
+++ b/app/src/testDemo/kotlin/com/google/samples/apps/nowinandroid/ui/NiaAppScreenSizesScreenshotTests.kt
@@ -141,18 +141,16 @@ class NiaAppScreenSizesScreenshotTests {
                 LocalInspectionMode provides true,
             ) {
                 DeviceConfigurationOverride(
-                    DeviceConfigurationOverride.ForcedSize(DpSize(width, height)),
+                    override = DeviceConfigurationOverride.ForcedSize(DpSize(width, height)),
                 ) {
-                    BoxWithConstraints {
-                        NiaTheme {
-                            NiaApp(
-                                windowSizeClass = WindowSizeClass.calculateFromSize(
-                                    DpSize(maxWidth, maxHeight),
-                                ),
-                                networkMonitor = networkMonitor,
-                                userNewsResourceRepository = userNewsResourceRepository,
-                            )
-                        }
+                    NiaTheme {
+                        NiaApp(
+                            windowSizeClass = WindowSizeClass.calculateFromSize(
+                                DpSize(width, height),
+                            ),
+                            networkMonitor = networkMonitor,
+                            userNewsResourceRepository = userNewsResourceRepository,
+                        )
                     }
                 }
             }