diff --git a/frigate/util/classification.py b/frigate/util/classification.py
index 3cd9a2b70..ab17a9444 100644
--- a/frigate/util/classification.py
+++ b/frigate/util/classification.py
@@ -75,6 +75,7 @@ class ClassificationTrainingProcess(FrigateProcess):
         logger.info(f"Kicking off classification training for {self.model_name}.")
         dataset_dir = os.path.join(CLIPS_DIR, self.model_name, "dataset")
         model_dir = os.path.join(MODEL_CACHE_DIR, self.model_name)
+        os.makedirs(model_dir, exist_ok=True)
         num_classes = len(
             [
                 d
diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 07a1db455..372c47429 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -67,7 +67,15 @@ export default function ModelSelectionView({
   }
 
   if (classificationConfigs.length == 0) {
-    return <NoModelsView onCreateModel={() => setNewModel(true)} />;
+    return (
+      <>
+        <ClassificationModelWizardDialog
+          open={newModel}
+          onClose={() => setNewModel(false)}
+        />
+        <NoModelsView onCreateModel={() => setNewModel(true)} />;
+      </>
+    );
   }
 
   return (