diff --git a/lib/flutter_image_picker.dart b/lib/flutter_image_picker.dart
index 5bd5c91..f503bfd 100644
--- a/lib/flutter_image_picker.dart
+++ b/lib/flutter_image_picker.dart
@@ -1,4 +1,4 @@
 library flutter_image_picker;
 
-export 'src/models/flutter_image_picker_theme.dart';
-export 'src/ui/flutter_image_picker_ui.dart';
+export 'src/models/image_picker_theme.dart';
+export 'src/ui/image_picker.dart';
diff --git a/test/mocks/image_picker_service_mock.dart b/test/mocks/image_picker_service_mock.dart
index 15256f7..31b7722 100644
--- a/test/mocks/image_picker_service_mock.dart
+++ b/test/mocks/image_picker_service_mock.dart
@@ -1,4 +1,4 @@
-import 'package:flutter_image_picker/src/services/flutter_image_picker_service.dart';
+import 'package:flutter_image_picker/src/services/image_picker_service.dart';
 import 'package:mocktail/mocktail.dart';
 
 class ImagePickerServiceMock extends Mock implements ImagePickerService {}