diff --git a/src/sparseml/version.py b/src/sparseml/version.py
index c03409d364b..21ccc54ec43 100644
--- a/src/sparseml/version.py
+++ b/src/sparseml/version.py
@@ -19,7 +19,7 @@
 from datetime import date
 
 
-version_base = "1.5.0"
+version_base = "1.5.1"
 is_release = False  # change to True to set the generated version as a release version
 
 
diff --git a/src/sparseml/yolov8/default.yaml b/src/sparseml/yolov8/default.yaml
index bfcfa9acb8d..edbba465b9b 100644
--- a/src/sparseml/yolov8/default.yaml
+++ b/src/sparseml/yolov8/default.yaml
@@ -2,6 +2,7 @@
 
 recipe: null
 recipe_args: null
+datasets_dir: null
 
 task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.
 mode: "train" # choices=['train', 'val', 'predict'] # mode to run task in.