Fixed the DB connection thing