diff --git a/dbus-service.py b/dbus-service.py
index 09cff5b..38532e2 100644
--- a/dbus-service.py
+++ b/dbus-service.py
@@ -6,6 +6,8 @@ from pydbus.generic import signal
 import pkg_resources
 from time import sleep
 from prototype import *
+from db97 import subtype_to_string
+from sensor import cancel_capture
 import pwd
 
 print("Starting up")
diff --git a/prototype.py b/prototype.py
index 7edf880..81f9261 100644
--- a/prototype.py
+++ b/prototype.py
@@ -5,6 +5,7 @@ from struct import pack, unpack
 from binascii import hexlify, unhexlify
 from tls97 import tls
 from usb97 import usb, unhex
+from sensor import enroll, identify
 from db97 import db
 from flash import read_flash
 from sid import *