diff --git a/src/com/android/car/media/browse/LimitedBrowseAdapter.java b/src/com/android/car/media/browse/LimitedBrowseAdapter.java
index 9189a16..bbca980 100644
--- a/src/com/android/car/media/browse/LimitedBrowseAdapter.java
+++ b/src/com/android/car/media/browse/LimitedBrowseAdapter.java
@@ -16,13 +16,11 @@
 
 package com.android.car.media.browse;
 
-import com.android.car.media.R;
-
-import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
 import androidx.recyclerview.widget.GridLayoutManager;
 import androidx.recyclerview.widget.RecyclerView;
 
+import com.android.car.media.R;
 import com.android.car.media.common.MediaItemMetadata;
 import com.android.car.ui.recyclerview.DelegatingContentLimitingAdapter;
 
@@ -62,7 +60,8 @@
                         return mMaxSpanSize;
                     }
 
-                    return mBrowseAdapter.getSpanSize(position, mMaxSpanSize);
+                    int itemIndex = positionToIndex(position);
+                    return mBrowseAdapter.getSpanSize(itemIndex, mMaxSpanSize);
                 }
             };