Fixing query builder tests.