diff --git a/i18n_sso.module b/i18n_sso.module
index b51121a..e05bade 100644
--- a/i18n_sso.module
+++ b/i18n_sso.module
@@ -47,7 +47,7 @@ function i18n_sso_preprocess_page(array &$variables): void {
  *
  * Removes all tokens for given account.
  */
-function i18n_sso_user_logout(AccountProxyInterface $account): void {
+function i18n_sso_user_logout(Drupal\user\Entity\User $account): void {
   \Drupal::service('i18n_sso.token')->deleteUserTokens($account->id());
 }