Added missing use.
Cleaning up query builder