diff --git a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/DragEventTrackerTest.kt b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/DragEventTrackerTest.kt
index 810eca4..df5d361 100644
--- a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/DragEventTrackerTest.kt
+++ b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/DragEventTrackerTest.kt
@@ -16,7 +16,7 @@
 
 package androidx.compose.foundation.text
 
-import androidx.ui.geometry.Offset
+import androidx.compose.ui.geometry.Offset
 import org.junit.Assert.assertEquals
 import org.junit.Test
 import org.junit.runner.RunWith
diff --git a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/TextFieldDelegateTest.kt b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/TextFieldDelegateTest.kt
index e190488..97eaf78 100644
--- a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/TextFieldDelegateTest.kt
+++ b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/TextFieldDelegateTest.kt
@@ -19,8 +19,8 @@
 import androidx.ui.core.Constraints
 import androidx.ui.core.LayoutCoordinates
 import androidx.ui.core.LayoutDirection
-import androidx.ui.geometry.Offset
-import androidx.ui.geometry.Rect
+import androidx.compose.ui.geometry.Offset
+import androidx.compose.ui.geometry.Rect
 import androidx.ui.graphics.Canvas
 import androidx.ui.graphics.Color
 import androidx.compose.ui.text.input.CommitTextEditOp
diff --git a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/SelectionModeTest.kt b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/SelectionModeTest.kt
index 4fc23ec..c71c3c7 100644
--- a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/SelectionModeTest.kt
+++ b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/SelectionModeTest.kt
@@ -17,8 +17,8 @@
 package androidx.compose.foundation.text.selection
 
 import androidx.test.filters.SmallTest
-import androidx.ui.geometry.Rect
-import androidx.ui.geometry.Offset
+import androidx.compose.ui.geometry.Rect
+import androidx.compose.ui.geometry.Offset
 import com.google.common.truth.Truth.assertThat
 import org.junit.Test
 import org.junit.runner.RunWith
diff --git a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/TextFieldSelectionManagerTest.kt b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/TextFieldSelectionManagerTest.kt
index 372dc4c..940059d 100644
--- a/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/TextFieldSelectionManagerTest.kt
+++ b/ui/ui-text/src/test/kotlin/androidx/compose/foundation/text/selection/TextFieldSelectionManagerTest.kt
@@ -23,8 +23,8 @@
 import androidx.ui.core.hapticfeedback.HapticFeedback
 import androidx.ui.core.hapticfeedback.HapticFeedbackType
 import androidx.ui.core.texttoolbar.TextToolbar
-import androidx.ui.geometry.Rect
-import androidx.ui.geometry.Offset
+import androidx.compose.ui.geometry.Rect
+import androidx.compose.ui.geometry.Offset
 import androidx.compose.ui.text.input.OffsetMap
 import androidx.compose.ui.text.input.TextFieldValue
 import androidx.compose.ui.text.AnnotatedString