diff --git a/app/(protected)/project/[project_id]/datasets/dataset/[dataset_id]/page.tsx b/app/(protected)/project/[project_id]/datasets/dataset/[dataset_id]/page.tsx
index 4cd9c82e..a34a88cd 100644
--- a/app/(protected)/project/[project_id]/datasets/dataset/[dataset_id]/page.tsx
+++ b/app/(protected)/project/[project_id]/datasets/dataset/[dataset_id]/page.tsx
@@ -96,7 +96,7 @@ export default function Dataset() {
               <ChevronLeft className="mr-1" />
               Back
             </Button>
-            <CreateData datasetId={dataset_id} />
+            <CreateData projectId={projectId} datasetId={dataset_id} />
             <DownloadDataset
               projectId={projectId}
               datasetId={dataset_id}
diff --git a/components/project/dataset/create-data.tsx b/components/project/dataset/create-data.tsx
index 911db76c..d1da4358 100644
--- a/components/project/dataset/create-data.tsx
+++ b/components/project/dataset/create-data.tsx
@@ -28,11 +28,13 @@ import { z } from "zod";
 
 export function CreateData({
   datasetId,
+  projectId,
   disabled = false,
   variant = "default",
   className = "",
 }: {
   datasetId?: string;
+  projectId?: string;
   disabled?: boolean;
   variant?: any;
   className?: string;
@@ -88,6 +90,7 @@ export function CreateData({
                       },
                     ],
                     datasetId,
+                    projectId,
                   }),
                 });
                 await queryClient.invalidateQueries(datasetId);