diff --git a/ai_edge_torch/generative/test/test_model_conversion_large.py b/ai_edge_torch/generative/test/test_model_conversion_large.py
index d3c6eb4b..fb229e6a 100644
--- a/ai_edge_torch/generative/test/test_model_conversion_large.py
+++ b/ai_edge_torch/generative/test/test_model_conversion_large.py
@@ -119,7 +119,7 @@ def test_phi2(self):
   )
   def test_phi3(self):
     config = phi3.get_fake_model_config()
-    pytorch_model = phi3.Phi3(config).eval()
+    pytorch_model = phi3.Phi3_5Mini(config).eval()
     self._test_model(
         config, pytorch_model, "prefill", atol=1e-5, rtol=1e-5
     )