Maybe move this to `Query` object?