diff --git a/tests/python/test_validate_country.py b/tests/python/test_validate_country.py
index ddbd27a5..b162e4df 100644
--- a/tests/python/test_validate_country.py
+++ b/tests/python/test_validate_country.py
@@ -1,5 +1,5 @@
 from flask import Response
-from policyengine_api.helpers import validate_country
+from policyengine_api.utils.payload_validators import validate_country
 
 
 @validate_country