From 228c4cdf9363bd7445bdb11311cb170ac6e5f257 Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Mon, 15 Jul 2024 07:26:07 -0700
Subject: [PATCH] add handling for 90 day email check

---
 app/notify_client/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app/notify_client/__init__.py b/app/notify_client/__init__.py
index 384c2ee3d..d6a11b78d 100644
--- a/app/notify_client/__init__.py
+++ b/app/notify_client/__init__.py
@@ -60,7 +60,7 @@ class NotifyAdminAPIClient(BaseAPIClient):
         still_signing_in = False
         for arg in args:
             arg = str(arg)
-            if "get-login-gov-user" in arg or "user/email" in arg or "/activate" in arg:
+            if "get-login-gov-user" in arg or "user/email" in arg or "/activate" or "/email-code" in arg:
                 still_signing_in = True
         # TODO:  Update this once E2E tests are managed by a feature flag or some other main config option.
         if os.getenv("NOTIFY_E2E_TEST_EMAIL"):