diff --git a/tests/test_ml.py b/tests/test_ml.py
index 46f36e50a..10b0441db 100644
--- a/tests/test_ml.py
+++ b/tests/test_ml.py
@@ -26,7 +26,7 @@
 
 BASE_URL = 'https://firebaseml.googleapis.com/v1beta2/'
 HEADER_CLIENT_KEY = 'X-FIREBASE-CLIENT'
-HEADER_CLIENT_VALUE = 'fire-admin-python/3.2.1'
+HEADER_CLIENT_VALUE = 'fire-admin-python/{0}'.format(firebase_admin.__version__)
 PROJECT_ID = 'my-project-1'
 
 PAGE_TOKEN = 'pageToken'