1 files changed, 4 insertions, 2 deletions
diff --git a/lib/view_model/badge_motive_view_model.dart b/lib/view_model/badge_motive_view_model.dart
index 6746827..75bd5b0 100644
--- a/lib/view_model/badge_motive_view_model.dart
+++ b/lib/view_model/badge_motive_view_model.dart
@@ -17,15 +17,17 @@ import 'package:flutter/foundation.dart';
 import 'package:logger/logger.dart';
 import 'package:uvok_epaper_badge/model/badge_motive.dart';
 import 'package:uvok_epaper_badge/model/motive_selection/badge_motive_selection.dart';
-import 'package:uvok_epaper_badge/model/motive_selection/mock_badge_motive_selection.dart';
 
 Logger logger = Logger();
 
 class BadgeMotiveViewModel extends ChangeNotifier {
-  final BadgeMotiveSelection _motivSelect = MockBadgeMotiveSelection();
+  final BadgeMotiveSelection _motivSelect;
   List<BadgeMotive> _motives = [];
   bool _busy = false;
 
+  BadgeMotiveViewModel({required BadgeMotiveSelection motivSelect})
+    : _motivSelect = motivSelect;
+
   bool get allowSelection => !_busy;
   List<BadgeMotive> get motives => _motives;