diff --git a/tests/test_config.py b/tests/test_config.py
index 60b5ab7..72733f8 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -19,7 +19,7 @@
 
 
 def test_get_model_config_by_name():
-    model_name = "facebook/opt-125m"
+    model_name = "facebook_opt-125m"
     model_config = get_model_config_by_name(model_name)
     assert isinstance(model_config, ModelConfig)
     assert model_config.num_layers == 12
diff --git a/tests/test_inference.py b/tests/test_inference.py
index 34ed6f0..42e11a2 100644
--- a/tests/test_inference.py
+++ b/tests/test_inference.py
@@ -52,7 +52,7 @@ def test_fastertransformer_13b_tp1():
 
 
 def test_llama2_70b():
-    model_name = "upstage/Llama-2-70b-instruct-v2"
+    model_name = "upstage_Llama-2-70b-instruct-v2"
     dtype_name = "w16a16e16"
     gpu_name = "a100-sxm-80gb"