diff --git a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py
index 0a65a82f7b..5b3706cc7c 100644
--- a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py
+++ b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py
@@ -18,11 +18,11 @@
 
 #  [START aiplatform_sdk_get_experiments_data_frame_sample]
 def get_experiments_data_frame_sample(
-    experiment_name: str,
+    experiment: str,
     project: str,
     location: str,
 ):
-    aiplatform.init(experiment_name=experiment_name, project=project, location=location)
+    aiplatform.init(experiment=experiment, project=project, location=location)
 
     experiments_df = aiplatform.get_experiment_df()
 
diff --git a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py
index 8740f5c2c7..b042de0bfc 100644
--- a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py
+++ b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py
@@ -22,7 +22,7 @@
 @pytest.mark.usefixtures("mock_sdk_init")
 def test_get_experiments_data_frame_sample(mock_get_experiment_df, mock_df):
     df = get_experiment_data_frame_sample.get_experiments_data_frame_sample(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
     )