Remove hardcode in getting DB connection.