diff --git a/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/build.gradle.kts b/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/build.gradle.kts
index 909387e0..8076aade 100644
--- a/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/build.gradle.kts
+++ b/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/build.gradle.kts
@@ -29,12 +29,12 @@ plugins {
 
 android {
   namespace = "com.google.ai.edge.examples.image_segmentation"
-  compileSdk = 34
+  compileSdk = 36
 
   defaultConfig {
     applicationId = "com.google.ai.edge.examples.image_segmentation"
     minSdk = 23
-    targetSdk = 33
+    targetSdk = 36
     versionCode = 1
     versionName = "1.0"
 
diff --git a/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/src/main/java/com/google/ai/edge/examples/image_segmentation/view/GalleryScreen.kt b/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/src/main/java/com/google/ai/edge/examples/image_segmentation/view/GalleryScreen.kt
index 4c0c20f8..c13e867f 100644
--- a/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/src/main/java/com/google/ai/edge/examples/image_segmentation/view/GalleryScreen.kt
+++ b/compiled_model_api/image_segmentation/kotlin_cpu_gpu/android/app/src/main/java/com/google/ai/edge/examples/image_segmentation/view/GalleryScreen.kt
@@ -103,7 +103,7 @@ fun GalleryScreen(
                     // Use the imageLoader extension property on Context
                     imageLoader = LocalContext.current.imageLoader,
                     onSuccess = { result ->
-                        val coilImage = (result as SuccessResult).image
+                        val coilImage = (result.result).image
                         val drawable = coilImage.asDrawable(context.resources)
                         val bitmap = (drawable as? android.graphics.drawable.BitmapDrawable)?.bitmap