diff --git a/app/src/main/java/com/hippo/widget/recyclerview/EasyRecyclerView.java b/app/src/main/java/com/hippo/widget/recyclerview/EasyRecyclerView.java
index 5cebf71c0..b29f4d0f1 100644
--- a/app/src/main/java/com/hippo/widget/recyclerview/EasyRecyclerView.java
+++ b/app/src/main/java/com/hippo/widget/recyclerview/EasyRecyclerView.java
@@ -734,13 +734,6 @@ public class EasyRecyclerView extends RecyclerView {
         return handled;
     }
 
-    @Override
-    protected void onLayout(boolean changed, int l, int t, int r, int b) {
-        super.onLayout(changed, l, t, r, b);
-
-        updateOnScreenCheckedViews();
-    }
-
     /**
      * Register a callback to be invoked when an item in this AdapterView has
      * been clicked.
@@ -759,6 +752,13 @@ public class EasyRecyclerView extends RecyclerView {
         mOnItemLongClickListener = listener;
     }
 
+    /**
+     * If choice mode is set, call it in {@link Adapter#onBindViewHolder}
+     */
+    public void checkItemCheckedState(View view, int position) {
+        setViewChecked(view, isItemChecked(position));
+    }
+
     /*
     @Override
     public Parcelable onSaveInstanceState() {