diff --git a/tests/integration/test_redis_relation.py b/tests/integration/test_redis_relation.py
index 339a26c..5535846 100644
--- a/tests/integration/test_redis_relation.py
+++ b/tests/integration/test_redis_relation.py
@@ -15,7 +15,6 @@
     check_application_status,
     get_address,
     get_unit_map,
-    get_unit_number,
     query_url,
 )