1 files changed, 6 insertions, 0 deletions
diff --git a/lib/model/device_connection_factory.dart b/lib/model/device_connection_factory.dart
index 23a186e..d73db0b 100644
--- a/lib/model/device_connection_factory.dart
+++ b/lib/model/device_connection_factory.dart
@@ -15,10 +15,16 @@
 
 import 'package:uvok_epaper_badge/model/device.dart';
 import 'package:uvok_epaper_badge/model/device_connection.dart';
+import 'package:uvok_epaper_badge/model/flutter_blue_plus_device.dart';
+import 'package:uvok_epaper_badge/model/flutter_blue_plus_device_connection.dart';
 import 'package:uvok_epaper_badge/model/mock_device_connection.dart';
 
 class DeviceConnectionFactory {
   static DeviceConnection createConnection(Device device) {
+    if (device is FlutterBluePlusDevice) {
+      return FlutterBluePlusDeviceConnection(device: device);
+    }
+
     // switch (device.type) {
     //   case DeviceType.ble:
     // return BleDeviceConnection(device);