diff --git a/README.md b/README.md
index 7115a7f..75a37bb 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@ On Ubuntu system:
 $ sudo apt remove fprintd
 $ sudo add-apt-repository ppa:uunicorn/open-fprintd
 $ sudo aptget update
-$ sudo apt install open-fprintd fprintd-clients python-validity
+$ sudo apt install open-fprintd fprintd-clients python3-validity
 ...wait a bit...
 $ fprintd-enroll
 ```
diff --git a/validitysensor/init.py b/validitysensor/init.py
index 6456de6..29508a4 100644
--- a/validitysensor/init.py
+++ b/validitysensor/init.py
@@ -3,8 +3,7 @@ import atexit
 
 from validitysensor.usb import usb
 from validitysensor.tls import tls
-from validitysensor.sensor import sensor
-from validitysensor.sensor import reboot
+from validitysensor.sensor import sensor, reboot, RebootException
 from validitysensor.flash import read_tls_flash
 from validitysensor.init_flash import init_flash
 from validitysensor.upload_fwext import upload_fwext