diff --git a/models/user_auth_token.go b/models/user_auth_token.go
index 4f9b679..b18c891 100644
--- a/models/user_auth_token.go
+++ b/models/user_auth_token.go
@@ -37,6 +37,9 @@ func GenerateUserAuthToken(ctx *gin.Context, verify security.Verifier, validityP
 	}
 	auth := *middleware.Auth
 	auth.Timeout = validityPeriod
+	auth.TimeoutFunc = func(_ interface{}) time.Duration {
+		return validityPeriod
+	}
 	userAuthToken := &UserAuthToken{
 		UserID: verify.GetUserID(),
 	}