diff --git a/examples/falling-balls/build.gradle.kts b/examples/falling-balls/build.gradle.kts
index f4bd464ec5..72a57467e5 100644
--- a/examples/falling-balls/build.gradle.kts
+++ b/examples/falling-balls/build.gradle.kts
@@ -6,7 +6,7 @@ plugins {
     // __KOTLIN_COMPOSE_VERSION__
     kotlin("jvm") version "1.5.31"
     // __LATEST_COMPOSE_RELEASE_VERSION__
-    id("org.jetbrains.compose") version "1.0.0-alpha4-build361"
+    id("org.jetbrains.compose") version "0.0.0-master-build413"
 }
 
 group = "me.user"
diff --git a/examples/falling-balls/src/main/kotlin/main.kt b/examples/falling-balls/src/main/kotlin/main.kt
index 8b24707295..0d9259f310 100644
--- a/examples/falling-balls/src/main/kotlin/main.kt
+++ b/examples/falling-balls/src/main/kotlin/main.kt
@@ -1,14 +1,14 @@
 package org.jetbrains.compose.demo.falling
 
 import androidx.compose.ui.ExperimentalComposeUiApi
+import androidx.compose.ui.unit.DpSize
 import androidx.compose.ui.unit.dp
-import androidx.compose.ui.window.WindowSize
 import androidx.compose.ui.window.WindowState
 import androidx.compose.ui.window.singleWindowApplication
 
 @OptIn(ExperimentalComposeUiApi::class)
 fun main() = singleWindowApplication(
-    title = "Falling Balls", state = WindowState(size = WindowSize(800.dp, 800.dp))
+    title = "Falling Balls", state = WindowState(size = DpSize(800.dp, 800.dp))
 ) {
     FallingBallsGame()
 }