diff --git a/tests/neural_compressor/test_optimization.py b/tests/neural_compressor/test_optimization.py
index f28c720138..9a1dc14f80 100644
--- a/tests/neural_compressor/test_optimization.py
+++ b/tests/neural_compressor/test_optimization.py
@@ -77,7 +77,7 @@ class OptimizationTest(INCTestMixin):
     )
 
     TEXT_GENERATION_SUPPORTED_ARCHITECTURES = (
-        "hf-internal-testing/tiny-random-BloomForCausalLM",
+        "hf-tiny-model-private/tiny-random-BloomForCausalLM",
         "hf-internal-testing/tiny-random-GPTNeoForCausalLM",
     )