diff --git a/tests/e2e/test_source_oauth.py b/tests/e2e/test_source_oauth.py
index 3f6ea6114..5bc0151be 100644
--- a/tests/e2e/test_source_oauth.py
+++ b/tests/e2e/test_source_oauth.py
@@ -25,6 +25,7 @@ from authentik.providers.oauth2.generators import (
 from authentik.sources.oauth.models import OAuthSource
 from authentik.sources.oauth.types.manager import SourceType
 from authentik.sources.oauth.types.twitter import TwitterOAuthCallback
+from authentik.stages.identification.models import IdentificationStage
 from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry
 
 CONFIG_PATH = "/tmp/dex.yml"  # nosec
@@ -115,7 +116,7 @@ class TestSourceOAuth2(SeleniumTestCase):
         authentication_flow = Flow.objects.get(slug="default-source-authentication")
         enrollment_flow = Flow.objects.get(slug="default-source-enrollment")
 
-        OAuthSource.objects.create(  # nosec
+        source = OAuthSource.objects.create(  # nosec
             name="dex",
             slug="dex",
             authentication_flow=authentication_flow,
@@ -127,6 +128,9 @@ class TestSourceOAuth2(SeleniumTestCase):
             consumer_key="example-app",
             consumer_secret=self.client_secret,
         )
+        ident_stage = IdentificationStage.objects.first()
+        ident_stage.sources.set([source])
+        ident_stage.save()
 
     @retry()
     @apply_migration("authentik_core", "0003_default_user")
@@ -308,7 +312,7 @@ class TestSourceOAuth1(SeleniumTestCase):
         authentication_flow = Flow.objects.get(slug="default-source-authentication")
         enrollment_flow = Flow.objects.get(slug="default-source-enrollment")
 
-        OAuthSource.objects.create(  # nosec
+        source = OAuthSource.objects.create(  # nosec
             name="oauth1",
             slug=self.source_slug,
             authentication_flow=authentication_flow,
@@ -317,6 +321,9 @@ class TestSourceOAuth1(SeleniumTestCase):
             consumer_key=self.client_id,
             consumer_secret=self.client_secret,
         )
+        ident_stage = IdentificationStage.objects.first()
+        ident_stage.sources.set([source])
+        ident_stage.save()
 
     @retry()
     @apply_migration("authentik_core", "0003_default_user")
diff --git a/tests/e2e/test_source_saml.py b/tests/e2e/test_source_saml.py
index 5673e4dde..ad8ee1930 100644
--- a/tests/e2e/test_source_saml.py
+++ b/tests/e2e/test_source_saml.py
@@ -16,6 +16,7 @@ from authentik.core.models import User
 from authentik.crypto.models import CertificateKeyPair
 from authentik.flows.models import Flow
 from authentik.sources.saml.models import SAMLBindingTypes, SAMLSource
+from authentik.stages.identification.models import IdentificationStage
 from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry
 
 LOGGER = get_logger()
@@ -119,7 +120,7 @@ class TestSourceSAML(SeleniumTestCase):
             key_data=IDP_KEY,
         )
 
-        SAMLSource.objects.create(
+        source = SAMLSource.objects.create(
             name="saml-idp-test",
             slug="saml-idp-test",
             authentication_flow=authentication_flow,
@@ -130,6 +131,9 @@ class TestSourceSAML(SeleniumTestCase):
             binding_type=SAMLBindingTypes.REDIRECT,
             signing_kp=keypair,
         )
+        ident_stage = IdentificationStage.objects.first()
+        ident_stage.sources.set([source])
+        ident_stage.save()
 
         self.driver.get(self.live_server_url)
 
@@ -199,6 +203,9 @@ class TestSourceSAML(SeleniumTestCase):
             binding_type=SAMLBindingTypes.POST,
             signing_kp=keypair,
         )
+        ident_stage = IdentificationStage.objects.first()
+        ident_stage.sources.set([source])
+        ident_stage.save()
 
         self.driver.get(self.live_server_url)
 
@@ -270,7 +277,7 @@ class TestSourceSAML(SeleniumTestCase):
             key_data=IDP_KEY,
         )
 
-        SAMLSource.objects.create(
+        source = SAMLSource.objects.create(
             name="saml-idp-test",
             slug="saml-idp-test",
             authentication_flow=authentication_flow,
@@ -281,6 +288,9 @@ class TestSourceSAML(SeleniumTestCase):
             binding_type=SAMLBindingTypes.POST_AUTO,
             signing_kp=keypair,
         )
+        ident_stage = IdentificationStage.objects.first()
+        ident_stage.sources.set([source])
+        ident_stage.save()
 
         self.driver.get(self.live_server_url)