Fix reference to db connection