Fix for db unit test