fixed up tests with the new mockDB