Add test for querying location