Fix test for single column entity in QueryGenerator