diff --git a/IQKeyBoardManager/IQKeyboardReturnKeyHandler.m b/IQKeyBoardManager/IQKeyboardReturnKeyHandler.m
index 9c35696..e4698ed 100644
--- a/IQKeyBoardManager/IQKeyboardReturnKeyHandler.m
+++ b/IQKeyBoardManager/IQKeyboardReturnKeyHandler.m
@@ -93,7 +93,7 @@ NSString *const kIQTextFieldReturnKeyType   =   @"kIQTextFieldReturnKeyType";
     {
         textField.keyboardType = [dict[kIQTextFieldReturnKeyType] integerValue];
         textField.delegate = dict[kIQTextFieldDelegate];
-        [textFieldInfoCache removeObject:textField];
+        [textFieldInfoCache removeObject:dict];
     }
 }
 
diff --git a/IQKeybordManagerSwift/IQKeyboardReturnKeyHandler.swift b/IQKeybordManagerSwift/IQKeyboardReturnKeyHandler.swift
index d44310b..a179bda 100644
--- a/IQKeybordManagerSwift/IQKeyboardReturnKeyHandler.swift
+++ b/IQKeybordManagerSwift/IQKeyboardReturnKeyHandler.swift
@@ -149,7 +149,7 @@ class IQKeyboardReturnKeyHandler: NSObject , UITextFieldDelegate, UITextViewDele
                 textView.delegate = dict[kIQTextFieldDelegate] as UITextViewDelegate?
             }
             
-            textFieldInfoCache.removeObject(view)
+            textFieldInfoCache.removeObject(dict)
         }
     }