diff --git a/simpeg_drivers/__init__.py b/simpeg_drivers/__init__.py
index 5c0152626..aeb8c67d7 100644
--- a/simpeg_drivers/__init__.py
+++ b/simpeg_drivers/__init__.py
@@ -131,7 +131,7 @@ def assets_path() -> Path:
     ),
     "joint surveys": (
         "simpeg_drivers.joint.joint_surveys",
-        {"inversion": "JointSurveyDriver"},
+        {"inversion": "JointSurveysDriver"},
     ),
     "magnetic scalar": (
         "simpeg_drivers.potential_fields.magnetic_scalar",
diff --git a/simpeg_drivers/joint/joint_cross_gradient/__init__.py b/simpeg_drivers/joint/joint_cross_gradient/__init__.py
index df32b2043..51b7e774f 100644
--- a/simpeg_drivers/joint/joint_cross_gradient/__init__.py
+++ b/simpeg_drivers/joint/joint_cross_gradient/__init__.py
@@ -7,3 +7,6 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
+
+from .driver import JointCrossGradientDriver
+from .options import JointCrossGradientOptions
diff --git a/simpeg_drivers/joint/joint_petrophysics/__init__.py b/simpeg_drivers/joint/joint_petrophysics/__init__.py
index df32b2043..4eee21f4e 100644
--- a/simpeg_drivers/joint/joint_petrophysics/__init__.py
+++ b/simpeg_drivers/joint/joint_petrophysics/__init__.py
@@ -7,3 +7,6 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
+
+from .driver import JointPetrophysicsDriver
+from .options import JointPetrophysicsOptions
diff --git a/simpeg_drivers/joint/joint_surveys/__init__.py b/simpeg_drivers/joint/joint_surveys/__init__.py
index df32b2043..782d0ab75 100644
--- a/simpeg_drivers/joint/joint_surveys/__init__.py
+++ b/simpeg_drivers/joint/joint_surveys/__init__.py
@@ -7,3 +7,6 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
+
+from .driver import JointSurveysDriver
+from .options import JointSurveysOptions
diff --git a/tests/uijson_test.py b/tests/uijson_test.py
index ee99b0033..2cfce2dd9 100644
--- a/tests/uijson_test.py
+++ b/tests/uijson_test.py
@@ -16,6 +16,7 @@
 import numpy as np
 import pytest
 from geoapps_utils.driver.data import BaseData
+from geoapps_utils.run import load_ui_json_as_dict
 from geoh5py import Workspace
 from geoh5py.ui_json import InputFile
 from geoh5py.ui_json.annotations import Deprecated
@@ -357,3 +358,11 @@ def test_legacy_uijson(tmp_path: Path, caplog):
                     assert "The Batch2D classes will be deprecated" in caplog.text
 
             assert driver.models
+
+
+def test_driver_from_uijson():
+    path = Path(__file__).resolve().parent / "legacy/v0.2.1"
+
+    for file in path.glob("*.ui.json"):
+        input_file = load_ui_json_as_dict(file)
+        assert from_input_file(input_file)