diff --git a/concordia/language_model/google_cloud_custom_model.py b/concordia/language_model/google_cloud_custom_model.py
index 9f751d8..fcea70f 100644
--- a/concordia/language_model/google_cloud_custom_model.py
+++ b/concordia/language_model/google_cloud_custom_model.py
@@ -108,7 +108,7 @@ def sample_text(
 
       try:
         response = (
-            aiplatform.Client()
+            aiplatform.PredictionServiceClient()
             .predict(
                 [{"content": prompt}],
                 parameters={