Fix the test case so that it works with other DBs