Update IdentityModule.cs