Fix DB connection