Fixed bug when setting customer query builder