Fix type typo about .query/.kudo_query refactoring