diff --git a/src/instructlab/training/main_ds.py b/src/instructlab/training/main_ds.py
index 63c92784..4c04da0f 100644
--- a/src/instructlab/training/main_ds.py
+++ b/src/instructlab/training/main_ds.py
@@ -775,7 +775,7 @@ def run_training(torch_args: TorchrunArgs, train_args: TrainingArgs) -> None:
     )
 
     # deepspeed options
-    if train_args.distributed_backend == DistributedBackend.DeepSpeed:
+    if train_args.distributed_backend == DistributedBackend.DEEPSPEED:
         if not FusedAdam:
             raise ImportError(
                 "DeepSpeed was selected as the distributed backend, but FusedAdam could not be imported. Please double-check that DeepSpeed is installed correctly"