DB connection fix