diff --git a/scripts/cleanup_datasets/admin_cleanup_datasets.py b/scripts/cleanup_datasets/admin_cleanup_datasets.py
index cee5d7916f3..1b2ba82e86c 100755
--- a/scripts/cleanup_datasets/admin_cleanup_datasets.py
+++ b/scripts/cleanup_datasets/admin_cleanup_datasets.py
@@ -53,6 +53,8 @@ from mako.template import Template
 from six.moves import configparser
 from sqlalchemy import and_, false
 
+sys.path.insert(1, os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, 'lib')))
+
 import galaxy.config
 import galaxy.model.mapping
 import galaxy.util
diff --git a/scripts/cleanup_datasets/update_dataset_size.py b/scripts/cleanup_datasets/update_dataset_size.py
index 177c4cb8337..dcf9267f1a5 100755
--- a/scripts/cleanup_datasets/update_dataset_size.py
+++ b/scripts/cleanup_datasets/update_dataset_size.py
@@ -10,6 +10,8 @@ import sys
 
 from six.moves import configparser
 
+sys.path.insert(1, os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, 'lib')))
+
 import galaxy.app
 
 assert sys.version_info[:2] >= (2, 4)
@@ -24,7 +26,7 @@ running.
 
 
 def main():
-    if len(sys.argv) != 1 or sys.argv[1] == "-h" or sys.argv[1] == "--help":
+    if len(sys.argv) != 2 or sys.argv[1] == "-h" or sys.argv[1] == "--help":
         usage(sys.argv[0])
         sys.exit()
     ini_file = sys.argv.pop(1)