Fix after sqlalchemy update