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"):