Fix to get DB