diff --git a/website/docs/users-sources/sources/social-logins/discord/index.md b/website/docs/users-sources/sources/social-logins/discord/index.md
index 1617192b5a..b76bcdd441 100644
--- a/website/docs/users-sources/sources/social-logins/discord/index.md
+++ b/website/docs/users-sources/sources/social-logins/discord/index.md
@@ -143,6 +143,8 @@ Ensure that the Discord OAuth source in **Federation & Social login** has the ad
 4. Provide a name for the policy and set the following expression:
 
 ```python
+from authentik.sources.oauth.models import OAuthSource
+
 # To get the guild ID number for the parameters, open Discord, go to Settings > Advanced and enable developer mode.
 # Right-click on the server/guild title and select "Copy ID" to get the guild ID.