Fixed import in QueryBuilder