Fix small bug in expected query results