diff --git a/daqviewer/daqviewer.py b/daqviewer/daqviewer.py
index 8c8b573..b2877be 100644
--- a/daqviewer/daqviewer.py
+++ b/daqviewer/daqviewer.py
@@ -16,7 +16,7 @@ from dash.dependencies import Input, Output
 import chart_studio.plotly as py
 import plotly.graph_objs as go
 
-import daqviewer.wafo.objects as wo
+import wafo.objects as wo
 
 app = dash.Dash()
 app.title = 'daqviewer'
diff --git a/setup.py b/setup.py
index 7aac1d8..e1a1c17 100644
--- a/setup.py
+++ b/setup.py
@@ -6,6 +6,7 @@ setuptools.setup(
     author='Dan Howe',
     author_email='d.howe@wrl.unsw.edu.au',
     description='Visualise DAQ csv files',
+    # packages=['daqviewer'],
     packages=setuptools.find_packages(),
     install_requires=[
         'pandas',