diff --git a/.ds.baseline b/.ds.baseline
index 98304c73d..69663da27 100644
--- a/.ds.baseline
+++ b/.ds.baseline
@@ -449,7 +449,7 @@
         "filename": "tests/__init__.py",
         "hashed_secret": "f8377c90fcfd699f0ddbdcb30c2c9183d2d933ea",
         "is_verified": false,
-        "line_number": 388,
+        "line_number": 389,
         "is_secret": false
       }
     ],
@@ -634,5 +634,5 @@
       }
     ]
   },
-  "generated_at": "2025-07-21T23:33:47Z"
+  "generated_at": "2025-07-22T17:07:31Z"
 }
diff --git a/tests/__init__.py b/tests/__init__.py
index 9a22f481e..98519744a 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -10,6 +10,7 @@ from flask import url_for
 from flask.testing import FlaskClient
 from flask_login import login_user
 
+from app.enums import ServicePermission
 from app.models.user import User
 
 # Add itsdangerous to the libraries which freezegun ignores to avoid errors.
diff --git a/tests/app/main/views/test_template_folders.py b/tests/app/main/views/test_template_folders.py
index f040301e6..f280a895a 100644
--- a/tests/app/main/views/test_template_folders.py
+++ b/tests/app/main/views/test_template_folders.py
@@ -3,6 +3,7 @@ import uuid
 import pytest
 from flask import abort, url_for
 
+from app.enums import ServicePermission
 from app.models.user import User
 from notifications_python_client.errors import HTTPError
 from tests import sample_uuid