diff --git a/examples/jetpack_compose/compose-ui/BUILD b/examples/jetpack_compose/compose-ui/BUILD
index 0320e4d13..bbab7117c 100644
--- a/examples/jetpack_compose/compose-ui/BUILD
+++ b/examples/jetpack_compose/compose-ui/BUILD
@@ -8,6 +8,7 @@ kt_android_library(
     plugins = ["//:jetpack_compose_compiler_plugin"],
     visibility = ["//visibility:public"],
     deps = [
+        "@maven//:androidx_activity_activity_compose",
         "@maven//:androidx_appcompat_appcompat",
         "@maven//:androidx_compose_material_material",
         "@maven//:androidx_compose_ui_ui",
diff --git a/examples/jetpack_compose/compose-ui/MainActivity.kt b/examples/jetpack_compose/compose-ui/MainActivity.kt
index c88efffb1..cefdca041 100644
--- a/examples/jetpack_compose/compose-ui/MainActivity.kt
+++ b/examples/jetpack_compose/compose-ui/MainActivity.kt
@@ -1,10 +1,10 @@
 package cm.ben.android.bazel.compose.example.ui
 
 import android.os.Bundle
+import androidx.activity.compose.setContent
 import androidx.appcompat.app.AppCompatActivity
 import androidx.compose.material.Text
 import androidx.compose.runtime.Composable
-import androidx.compose.ui.platform.setContent
 
 class MainActivity : AppCompatActivity() {
     override fun onCreate(savedInstanceState: Bundle?) {