1 files changed, 3 insertions, 7 deletions
diff --git a/ausbutler/db.py b/ausbutler/db.py
index 940c69d..e40fee5 100644
--- a/ausbutler/db.py
+++ b/ausbutler/db.py
@@ -1,15 +1,11 @@
-import json
-from os import path
-
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
-import __main__
+from .config import load_config
 
 def get_session():
     session = sessionmaker(bind=create_engine(
         "mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}".format(
-            json.load(open(
-                path.join(path.dirname(__main__.__file__), 'config', 'db.json')
-            )))))
+            load_config('db')
+        )))
     return session()