diff --git a/release/golden_notebook_tests/workloads/torch_tune_serve_test.py b/release/golden_notebook_tests/workloads/torch_tune_serve_test.py
index <HASH>..<HASH> 100644
--- a/release/golden_notebook_tests/workloads/torch_tune_serve_test.py
+++ b/release/golden_notebook_tests/workloads/torch_tune_serve_test.py
@@ -210,7 +210,7 @@ if __name__ == "__main__":
     analysis = train_mnist(args.smoke_test, num_workers, use_gpu)
 
     print("Retrieving best model.")
-    best_checkpoint = analysis.best_checkpoint
+    best_checkpoint = analysis.best_checkpoint.local_path
     model_id = get_remote_model(best_checkpoint)
 
     print("Setting up Serve.")
[release test] Update torch_tune_serve test to be compatible with new TrialCheckpoint class. (#<I>)