Add additional checks for DB connection loss