Updated the db test helper