diff --git a/lib/models/ai/chats_model.dart b/lib/models/ai/chats_model.dart
index 4651008..2accc8d 100644
--- a/lib/models/ai/chats_model.dart
+++ b/lib/models/ai/chats_model.dart
@@ -1,6 +1,5 @@
 import 'package:didvan/models/ai/bots_model.dart';
 import 'package:didvan/models/ai/files_model.dart';
-import 'package:image_picker/image_picker.dart';
 
 class ChatsModel {
   int? id;
diff --git a/lib/views/direct/direct.dart b/lib/views/direct/direct.dart
index b5d611a..804855a 100644
--- a/lib/views/direct/direct.dart
+++ b/lib/views/direct/direct.dart
@@ -5,7 +5,7 @@ import 'package:didvan/constants/assets.dart';
 import 'package:didvan/models/enums.dart';
 import 'package:didvan/models/view/app_bar_data.dart';
 import 'package:didvan/providers/server_data.dart';
-import 'package:didvan/services/media/media.dart';
+import 'package:didvan/services/media/voice.dart';
 import 'package:didvan/views/direct/direct_state.dart';
 import 'package:didvan/views/direct/widgets/message.dart';
 import 'package:didvan/views/direct/widgets/message_box.dart';
@@ -54,7 +54,7 @@ class _DirectState extends State<Direct> {
     final d = MediaQuery.of(context);
     return WillPopScope(
       onWillPop: () async {
-        MediaService.resetAudioPlayer();
+        VoiceService.resetVoicePlayer();
         return true;
       },
       child: Material(