diff --git a/studio/frontend/src/features/studio/sections/dataset-section.tsx b/studio/frontend/src/features/studio/sections/dataset-section.tsx
index 28254adcc7..8438e1a7fb 100644
--- a/studio/frontend/src/features/studio/sections/dataset-section.tsx
+++ b/studio/frontend/src/features/studio/sections/dataset-section.tsx
@@ -76,6 +76,8 @@ export function DatasetSection() {
     setDatasetEvalSplit,
     hfToken,
     modelType,
+    isVisionModel,
+    isCheckingVision,
     datasetSliceStart,
     setDatasetSliceStart,
     datasetSliceEnd,
@@ -94,6 +96,8 @@ export function DatasetSection() {
       setDatasetEvalSplit: s.setDatasetEvalSplit,
       hfToken: s.hfToken,
       modelType: s.modelType,
+      isVisionModel: s.isVisionModel,
+      isCheckingVision: s.isCheckingVision,
       datasetSliceStart: s.datasetSliceStart,
       setDatasetSliceStart: s.setDatasetSliceStart,
       datasetSliceEnd: s.datasetSliceEnd,
@@ -119,6 +123,9 @@ export function DatasetSection() {
     }
     setInputValue(val);
   }
+
+  const effectiveModelType = !isCheckingVision && isVisionModel ? "vision" : modelType;
+
   const {
     results: hfResults,
     isLoading,
@@ -126,7 +133,7 @@ export function DatasetSection() {
     fetchMore,
     error: hfSearchError,
   } = useHfDatasetSearch(debouncedQuery, {
-    modelType,
+    modelType: effectiveModelType,
     accessToken: hfToken || undefined,
   });