diff --git a/lambdas/services/mock_login_redirect_service.py b/lambdas/services/mock_login_redirect_service.py index 9c7c08fb2..875dd5f53 100644 --- a/lambdas/services/mock_login_redirect_service.py +++ b/lambdas/services/mock_login_redirect_service.py @@ -25,7 +25,7 @@ def prepare_redirect_response(self, event): state = "".join(random.choices(string.ascii_letters + string.digits, k=30)) self.save_state_in_dynamo_db(state) - if os.getenv("WORKSPACE") == "pre-prod": + if os.getenv("WORKSPACE") in ["pre-prod", "ndr-test"]: clean_url = re.sub(r"^api.", "", host) else: clean_url = re.sub(r"^api-", "", host)