Fixed issue with DB connection