Update queries_test.go