diff --git a/src/llama_stack_client/lib/cli/post_training/post_training.py b/src/llama_stack_client/lib/cli/post_training/post_training.py
index 3a890b80..2d766652 100644
--- a/src/llama_stack_client/lib/cli/post_training/post_training.py
+++ b/src/llama_stack_client/lib/cli/post_training/post_training.py
@@ -10,7 +10,7 @@
 from rich.console import Console
 
 from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import (
-    AlgorithmConfigParam,
+    AlgorithmConfig,
     TrainingConfig,
 )
 
@@ -36,7 +36,7 @@ def supervised_fine_tune(
     ctx,
     job_uuid: str,
     model: str,
-    algorithm_config: AlgorithmConfigParam,
+    algorithm_config: AlgorithmConfig,
     training_config: TrainingConfig,
     checkpoint_dir: Optional[str],
 ):