Flask-Login: AttributeError: type object ‘User’ has no attribute ‘get’

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(user_id)