Fix SQLAlchemy bug