diff --git a/predict_loading_concentration.py b/predict_loading_concentration.py
index 456021c..ff56abb 100755
--- a/predict_loading_concentration.py
+++ b/predict_loading_concentration.py
@@ -7,11 +7,11 @@ def warn(*args, **kwargs):
 from diva_seq_opt import utility
 import argparse
 import pickle
+import matplotlib as mpl
 
 
 def main():
 	#Allow Matplotlib to be used on commandline
-	import matplotlib as mpl
 	mpl.use('Agg')
 
 	#Handle inputs
diff --git a/setup.py b/setup.py
index 40cd3da..fe8b8ef 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='diva_seq_opt',
-      version='0.0.3',
+      version='0.0.4',
       description='Predict the library loading concentration to use for mi-seq runs based on bioanalyzer data.',
       url='https://github.com/JBEI/library_loading_concentration_prediction',
       author='Zak Costello',
@@ -9,11 +9,11 @@
       license='BSD',
       packages=['diva_seq_opt'],      
       entry_points = {
-        'console_scripts': ['predict_loading_concentration=diva_seq_opt.predict_loading_concentration:main'],
+        'console_scripts': ['predict_loading_concentration=predict_loading_concentration:main'],
       },
       install_requires=[
           'pandas','numpy','scipy','matplotlib',
           'peakutils','sklearn','tpot'
       ],
       include_package_data=True,
-      zip_safe=False)
\ No newline at end of file
+      zip_safe=False)