diff --git a/demo.py b/demo.py
index 62c2033..8efd67f 100755
--- a/demo.py
+++ b/demo.py
@@ -26,7 +26,7 @@ def choice(*values):
         return np.random.choice(values, size)
 
     random_dates = [
-        datetime.date(2016, 1, 1) + datetime.timedelta(days=delta)
+        datetime.date(2016, 1, 1) + datetime.timedelta(days=int(delta))
         for delta in np.random.randint(1, 50, size)
     ]
 
diff --git a/dfgui/dfgui.py b/dfgui/dfgui.py
index a505f6b..18e421e 100644
--- a/dfgui/dfgui.py
+++ b/dfgui/dfgui.py
@@ -3,7 +3,15 @@
 
 from __future__ import absolute_import, division, print_function
 
-import wx
+try:
+    import wx
+except ImportError:
+    import sys
+    sys.path += [
+        "/usr/lib/python2.7/dist-packages/wx-2.8-gtk2-unicode",
+        "/usr/lib/python2.7/dist-packages"
+    ]
+    import wx
 
 import matplotlib
 matplotlib.use('WXAgg')