make test database more flexible