Fix comment
Add full PIR query tests