diff --git a/plate_simulation-assets/uijson/plate_simulation.ui.json b/plate_simulation-assets/uijson/plate_simulation.ui.json
index e28c381..408e7a6 100644
--- a/plate_simulation-assets/uijson/plate_simulation.ui.json
+++ b/plate_simulation-assets/uijson/plate_simulation.ui.json
@@ -1,7 +1,7 @@
 {
+    "version": "0.2.0-beta.1",
+    "title": "Plate Simulation",
     "icon": "maxwellplate",
-    "version": "0.2.0-alpha.1",
-    "title": "Plate simulation",
     "documentation": "https://mirageoscience-plate-simulation.readthedocs-hosted.com/en/latest/",
     "conda_environment": "plate_simulation",
     "run_command": "plate_simulation.driver",
diff --git a/plate_simulation/options.py b/plate_simulation/options.py
index b090e9a..f8a79bb 100644
--- a/plate_simulation/options.py
+++ b/plate_simulation/options.py
@@ -91,7 +91,7 @@ def simulation_parameters(self) -> BaseForwardOptions:
             raise ValueError("Input file data must be set.")
 
         if input_file.data["inversion_type"] in PARAM_MAP:
-            return PARAM_MAP[input_file.data["inversion_type"]].build(input_file.data)
+            return PARAM_MAP[input_file.data["inversion_type"]].build(**input_file.data)
 
         raise NotImplementedError(
             f"Unknown inversion type: {input_file.data['inversion_type']}"
diff --git a/tests/runtest/gravity_test.py b/tests/runtest/gravity_test.py
index 0770688a..ec2e62b 100644
--- a/tests/runtest/gravity_test.py
+++ b/tests/runtest/gravity_test.py
@@ -10,6 +10,7 @@
 import numpy as np
 from geoh5py import Workspace
 from geoh5py.groups import SimPEGGroup
+from simpeg_drivers.options import ActiveCellsOptions
 from simpeg_drivers.potential_fields.gravity.options import GravityForwardOptions
 
 from plate_simulation.driver import PlateSimulationDriver
@@ -59,15 +60,14 @@ def test_gravity_plate_simulation(tmp_path):
             plate_model=plate_params,
         )
 
-        options = GravityForwardOptions.build(
-            topography_object=topography,
-            data_object=survey,
-            geoh5=ws,
-            starting_model=0.1,
+        active_cells = ActiveCellsOptions(topography_object=topography)
+        inputs = {"geoh5": ws, "active_cells": active_cells, "data_object": survey}
+        options = GravityForwardOptions.model_construct(
+            **inputs,
         )
 
-        gravity_inversion = SimPEGGroup.create(ws)
-        gravity_inversion.options = options.serialize()
+        gravity_forward = SimPEGGroup.create(ws)
+        gravity_forward.options = options.serialize()
 
         params = PlateSimulationOptions(
             title="test",
@@ -75,7 +75,7 @@ def test_gravity_plate_simulation(tmp_path):
             geoh5=ws,
             mesh=mesh_params,
             model=model_params,
-            simulation=gravity_inversion,
+            simulation=gravity_forward,
         )
         driver = PlateSimulationDriver(params)
         driver.run()