diff --git a/tests/test_omlflow.py b/tests/test_omlflow.py
index 8f5aa49..3e793ce 100644
--- a/tests/test_omlflow.py
+++ b/tests/test_omlflow.py
@@ -15,7 +15,7 @@ def _objective_func(trial):
             # do folds
             for i in range(num_folds):
                 result = (x - 2) ** 2
-                om.log_iter({"x": result}, i)
+                om.log_iter({"result": result}, i)
                 results.append(result)
 
             result = np.mean(results)
@@ -45,11 +45,8 @@ def test_study_name(tmpdir):
     run_infos = mlfl_client.list_run_infos(experiment_id)
     assert len(run_infos) == n_trials + n_trials * num_folds
 
-    first_run_id = run_infos[0].run_id
+    first_run_id = run_infos[-1].run_id
     first_run = mlfl_client.get_run(first_run_id)
     first_run_dict = first_run.to_dictionary()
     assert "x" in first_run_dict["data"]["params"]
-    assert "y" in first_run_dict["data"]["params"]
-    assert "z" in first_run_dict["data"]["params"]
     assert first_run_dict["data"]["tags"]["direction"] == "MINIMIZE"
-    assert first_run_dict["data"]["tags"]["state"] == "COMPLETE"