Fixes query count test.