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 b9b353fb..54c87a83 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
@@ -9,7 +9,7 @@
 import click
 from rich.console import Console
 
-from llama_stack_client.types.post_training_supervised_fine_tune_params import AlgorithmConfigParam, TrainingConfig
+from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import AlgorithmConfigParam, TrainingConfig
 
 from ..common.utils import handle_client_errors