diff --git a/lit_nlp/examples/gcp/model_server_test.py b/lit_nlp/examples/gcp/model_server_test.py
index 1874e72c..666fec0f 100644
--- a/lit_nlp/examples/gcp/model_server_test.py
+++ b/lit_nlp/examples/gcp/model_server_test.py
@@ -25,9 +25,9 @@ def test_predict_endpoint(self, mock_get_models):
     ]
 
     mock_get_models.return_value = {
-        'gemma_1.1_instruct_2b_en': mock_model,
-        'gemma_1.1_instruct_2b_en_salience': salience_model,
-        'gemma_1.1_instruct_2b_en_tokenize': tokenize_model,
+        'gemma_1.1_2b_IT': mock_model,
+        'gemma_1.1_2b_IT_salience': salience_model,
+        'gemma_1.1_2b_IT_tokenize': tokenize_model,
     }
     app = webtest.TestApp(model_server.get_wsgi_app())