diff --git a/models/cli/download.py b/models/cli/download.py
index 2fcb99126..7f0a92805 100644
--- a/models/cli/download.py
+++ b/models/cli/download.py
@@ -461,7 +461,7 @@ def run_download_cmd(args: argparse.Namespace, parser: argparse.ArgumentParser):
 
         from llama_models.sku_list import llama_meta_net_info, resolve_model
 
-        from .model.safety_models import (
+        from .safety_models import (
             prompt_guard_download_info_map,
             prompt_guard_model_sku_map,
         )