Fixed import issue for sqlalchemy