diff --git a/plot.py b/plot.py
index d41c493..104027a 100644
--- a/plot.py
+++ b/plot.py
@@ -21,7 +21,7 @@ def plot():
     Plots the data contained in the output file
     """
     logging.info('Plotting data..')
-    output_dir = hlp.get_output_directories()[0]
+    output_dir = hlp.get_output_directory()
 
     figures_path = output_dir / 'figures'
     if not figures_path.is_dir():
diff --git a/tokenomics_decentralization/helper.py b/tokenomics_decentralization/helper.py
index 78f84c3..0a23fba 100644
--- a/tokenomics_decentralization/helper.py
+++ b/tokenomics_decentralization/helper.py
@@ -144,13 +144,13 @@ def increment_date(date, by):
 def get_output_directory():
     """
     Reads the config file and retrieves the output directories
-    :returns: a list of directories that might contain the db files
+    :returns: the directory to use for the output files
     """
     config = get_config_data()
     sources = ' - '.join(get_active_source_keywords())
     if not sources:
         sources = 'No clustering'
-    return [pathlib.Path(db_dir).resolve() for db_dir in config['output_directories']][0] / sources
+    return pathlib.Path(config['output_directories'][0]).resolve() / sources
 
 
 def get_input_directories():