diff --git a/lib/screens/mind_map_screen.dart b/lib/screens/mind_map_screen.dart
index 1d3e1eb..ed0def6 100644
--- a/lib/screens/mind_map_screen.dart
+++ b/lib/screens/mind_map_screen.dart
@@ -1,6 +1,7 @@
 import 'package:flutter/material.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
 import '../providers/mindmap_provider.dart';
+import '../models/mind_map_model.dart';
 
 class MindMapScreen extends ConsumerStatefulWidget {
   const MindMapScreen({super.key});