diff --git a/ui/src/views/paragraph/component/ProblemComponent.vue b/ui/src/views/paragraph/component/ProblemComponent.vue
index cef42b4e1ee..6284f3285ef 100644
--- a/ui/src/views/paragraph/component/ProblemComponent.vue
+++ b/ui/src/views/paragraph/component/ProblemComponent.vue
@@ -168,7 +168,7 @@ const remoteMethod = (query: string) => {
 function getProblemOption(filterText?: string) {
   return problem
     .asyncGetProblem(
-      id as string,
+      props.datasetId || (id as string),
       { current_page: 1, page_size: 100 },
       filterText && { content: filterText },
       optionLoading