Fixed DB issue for Heroku