diff --git a/library/src/main/java/com/unnamed/b/atv/model/TreeNode.java b/library/src/main/java/com/unnamed/b/atv/model/TreeNode.java
index dbf33f8..190f18d 100644
--- a/library/src/main/java/com/unnamed/b/atv/model/TreeNode.java
+++ b/library/src/main/java/com/unnamed/b/atv/model/TreeNode.java
@@ -222,6 +222,7 @@ public interface TreeNodeLongClickListener {
     public static abstract class BaseNodeViewHolder<E> {
         protected AndroidTreeView tView;
         protected TreeNode mNode;
+        private View nodeView;
         private View mView;
         protected int containerStyle;
         protected Context context;
@@ -234,7 +235,7 @@ public View getView() {
             if (mView != null) {
                 return mView;
             }
-            final View nodeView = getNodeView();
+            nodeView = getNodeView();
             final TreeNodeWrapperView nodeWrapperView = new TreeNodeWrapperView(nodeView.getContext(), getContainerStyle());
             nodeWrapperView.insertNodeView(nodeView);
             mView = nodeWrapperView;
@@ -242,8 +243,16 @@ public View getView() {
             return mView;
         }
 
-        public void setTreeViev(AndroidTreeView treeViev) {
-            this.tView = treeViev;
+        public void setOnClickListener(View.OnClickListener listener) {
+            nodeView.setOnClickListener(listener);
+        }
+
+        public void setOnLongClickListener(View.OnLongClickListener listener) {
+            nodeView.setOnLongClickListener(listener);
+        }
+
+        public void setTreeView(AndroidTreeView treeView) {
+            this.tView = treeView;
         }
 
         public AndroidTreeView getTreeView() {
diff --git a/library/src/main/java/com/unnamed/b/atv/view/AndroidTreeView.java b/library/src/main/java/com/unnamed/b/atv/view/AndroidTreeView.java
index 222a43a..e14bbb0 100644
--- a/library/src/main/java/com/unnamed/b/atv/view/AndroidTreeView.java
+++ b/library/src/main/java/com/unnamed/b/atv/view/AndroidTreeView.java
@@ -261,7 +261,7 @@ private void addNode(ViewGroup container, final TreeNode n) {
             viewHolder.toggleSelectionMode(mSelectionModeEnabled);
         }
 
-        nodeView.setOnClickListener(new View.OnClickListener() {
+        viewHolder.setOnClickListener(new View.OnClickListener() {
             @Override
             public void onClick(View v) {
                 if (n.getClickListener() != null) {
@@ -275,7 +275,7 @@ public void onClick(View v) {
             }
         });
 
-        nodeView.setOnLongClickListener(new View.OnLongClickListener() {
+        viewHolder.setOnLongClickListener(new View.OnLongClickListener() {
             @Override
             public boolean onLongClick(View view) {
                 if (n.getLongClickListener() != null) {
@@ -408,7 +408,7 @@ private TreeNode.BaseNodeViewHolder getViewHolderForNode(TreeNode node) {
             viewHolder.setContainerStyle(containerStyle);
         }
         if (viewHolder.getTreeView() == null) {
-            viewHolder.setTreeViev(this);
+            viewHolder.setTreeView(this);
         }
         return viewHolder;
     }