diff --git a/ui/src/breezydesktop.in b/ui/src/breezydesktop.in
index 2729644..f65166b 100755
--- a/ui/src/breezydesktop.in
+++ b/ui/src/breezydesktop.in
@@ -21,6 +21,12 @@
 
 import os
 import sys
+import os
+import sys
+
+lib_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib')
+sys.path.insert(0, lib_dir)
+
 import signal
 import locale
 import gettext
diff --git a/ui/src/main.py b/ui/src/main.py
index e96cdcf..d12bbc8 100644
--- a/ui/src/main.py
+++ b/ui/src/main.py
@@ -19,10 +19,6 @@
 
 import os
 import sys
-
-lib_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib')
-sys.path.insert(0, lib_dir)
-
 import gi
 import logging
 import argparse
diff --git a/ui/src/virtualdisplay.py b/ui/src/virtualdisplay.py
index 64eb478..4ffd3f2 100644
--- a/ui/src/virtualdisplay.py
+++ b/ui/src/virtualdisplay.py
@@ -4,7 +4,7 @@ import argparse
 import logging
 import sys
 import signal
-from .lib import pydbus
+import pydbus
 import gi
 import time