Remove check for User model