Fix code to ensure db connection call works.