diff --git a/mlos_bench/mlos_bench/tests/storage/trial_data_test.py b/mlos_bench/mlos_bench/tests/storage/trial_data_test.py
index 77bc1eb243..62c7bfb1bf 100644
--- a/mlos_bench/mlos_bench/tests/storage/trial_data_test.py
+++ b/mlos_bench/mlos_bench/tests/storage/trial_data_test.py
@@ -21,7 +21,7 @@ def test_exp_trial_data(exp_data: ExperimentData) -> None:
     assert trial.tunable_config_id == expected_config_id
     assert trial.status == Status.SUCCEEDED
     assert trial.metadata_dict["repeat_i"] == 1
-    assert trial.metadata_dict["trial_runner_id"] == "1"
+    assert trial.trial_runner_id == 1
     assert list(trial.results_dict.keys()) == ["score"]
     assert trial.results_dict["score"] == pytest.approx(73.27, 0.01)
     assert isinstance(trial.ts_start, datetime)