diff --git a/dpadrecyclerview/src/androidTest/kotlin/com/rubensousa/dpadrecyclerview/test/tests/drag/DragHelperGridTest.kt b/dpadrecyclerview/src/androidTest/kotlin/com/rubensousa/dpadrecyclerview/test/tests/drag/DragHelperGridTest.kt
index da9de17a..7651e4bb 100644
--- a/dpadrecyclerview/src/androidTest/kotlin/com/rubensousa/dpadrecyclerview/test/tests/drag/DragHelperGridTest.kt
+++ b/dpadrecyclerview/src/androidTest/kotlin/com/rubensousa/dpadrecyclerview/test/tests/drag/DragHelperGridTest.kt
@@ -160,13 +160,12 @@ class DragHelperGridTest {
         KeyEvents.pressUp(times = 1)
 
         // then
-        assertFocusAndSelection(position = bottomColumnPosition)
+        assertFocusAndSelection(position = topColumnPosition)
         testAdapter.assertContents { index ->
             when {
                 index < topColumnPosition -> index
                 index == topColumnPosition -> bottomColumnPosition
-                index < bottomColumnPosition -> index - 1
-                index == bottomColumnPosition -> topColumnPosition
+                index <= bottomColumnPosition -> index - 1
                 else -> index
             }
         }