diff --git a/tests/app/conftest.py b/tests/app/conftest.py
index 23e7bcede..d48ba9069 100644
--- a/tests/app/conftest.py
+++ b/tests/app/conftest.py
@@ -9,7 +9,6 @@ from flask import current_app
 
 from app import db
 from app.models import (
-    User,
     Service,
     Template,
     ApiKey,
diff --git a/tests/app/delivery/test_send_to_providers.py b/tests/app/delivery/test_send_to_providers.py
index ee7948292..2894d75e9 100644
--- a/tests/app/delivery/test_send_to_providers.py
+++ b/tests/app/delivery/test_send_to_providers.py
@@ -23,7 +23,7 @@ from app.models import (
 from tests.app.db import create_service, create_template, create_notification
 
 
-def test_should_return_highest_priority_active_provider(notify_db_session):
+def test_should_return_highest_priority_active_provider(restore_provider_details):
     providers = provider_details_dao.get_provider_details_by_notification_type('sms')
 
     first = providers[0]