diff --git a/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/adapter/CustomAdapter.kt b/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/adapter/CustomAdapter.kt
index 427ebf9..3b28b6d 100644
--- a/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/adapter/CustomAdapter.kt
+++ b/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/adapter/CustomAdapter.kt
@@ -20,7 +20,7 @@ class CustomAdapter<T>(private val activity:AppCompatActivity?,private val isPag
     private var anyObject: Any? = null
     private var _layout: Int? = null
 
-    internal lateinit var smartRecyclerViewListener: SmartRecyclerViewListener<T>
+    lateinit var smartRecyclerViewListener: SmartRecyclerViewListener<T>
     internal lateinit var viewAttachListener: ViewAttachListener<T>
 
     override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): CustomViewHolder<T> {
diff --git a/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/viewholder/CustomViewHolder.kt b/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/viewholder/CustomViewHolder.kt
index 05f28e5..ce849ca 100644
--- a/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/viewholder/CustomViewHolder.kt
+++ b/smartrecyclerview/src/main/java/com/kunalapk/smartrecyclerview/viewholder/CustomViewHolder.kt
@@ -1,7 +1,6 @@
 package com.kunalapk.smartrecyclerview.viewholder
 
 import androidx.databinding.ViewDataBinding
-import androidx.lifecycle.ViewModel
 import androidx.recyclerview.widget.RecyclerView
 import com.kunalapk.smartrecyclerview.BR