diff --git a/azuremlpythonsdk-v2/azml_01_experiment_remote_compute.py b/azuremlpythonsdk-v2/azml_01_experiment_remote_compute.py
index cd89524..fd5e32c 100644
--- a/azuremlpythonsdk-v2/azml_01_experiment_remote_compute.py
+++ b/azuremlpythonsdk-v2/azml_01_experiment_remote_compute.py
@@ -20,7 +20,7 @@ registered_model_name = "diabetes_model"
 
 def main():
     # 1. Create or Load a ML client
-    ml_client = XXXX()
+    ml_client = create_or_load_ml_client()
 
     # 2. Create compute resources
     create_or_load_aml()
@@ -29,7 +29,7 @@ def main():
     create_tabular_dataset()
     latest_version_dataset = next(
         dataset.latest_version
-        for dataset in ml_client.data.XXXX
+        for dataset in ml_client.data.list()
         if dataset.name == name_dataset
     )
 
@@ -40,15 +40,15 @@ def main():
             data=Input(
                 type=AssetTypes.URI_FILE,
                 # @latest doesn't work with dataset paths
-                path=XXXX,
+                path=f"azureml:{name_dataset}:{latest_version_dataset}",
             ),
             registered_model_name=registered_model_name,
         ),
         code=experiment_folder,
         command=(
             "python ${{inputs.script_name}}"
-            + " --data XXXX"
-            + " --registered_model_name XXXX"
+            + " --data ${{inputs.data}}"
+            + " --registered_model_name ${{inputs.registered_model_name}}"
         ),
         environment=f"{custom_env_name}@latest",
         compute=AML_COMPUTE_NAME,