diff --git a/iOS/TUIKitDemo/TUIKitDemo/Chat/ChatViewController.m b/iOS/TUIKitDemo/TUIKitDemo/Chat/ChatViewController.m
index daa45e0ced6..5252acf7888 100644
--- a/iOS/TUIKitDemo/TUIKitDemo/Chat/ChatViewController.m
+++ b/iOS/TUIKitDemo/TUIKitDemo/Chat/ChatViewController.m
@@ -33,7 +33,6 @@
 #import "TCUtil.h"
 #import "THelper.h"
 #import "TCConstants.h"
-#import "V2TIMManager.h"
 #import "GenerateTestUserSig.h"
 #import "Toast.h"
 #import "TUIKitListenerManager.h"
@@ -50,6 +49,8 @@
 #import "TUILiveHeartBeatManager.h"
 #endif
 
+@import ImSDK_Plus;
+
 // MLeaksFinder 会对这个类误报，这里需要关闭一下
 @implementation UIImagePickerController (Leak)