Fix QueryBuilder method signature