diff --git a/web/app/components/datasets/metadata/metadata-dataset/create-metadata-modal.tsx b/web/app/components/datasets/metadata/metadata-dataset/create-metadata-modal.tsx
index 179ed9607a..6e00aa4b2c 100644
--- a/web/app/components/datasets/metadata/metadata-dataset/create-metadata-modal.tsx
+++ b/web/app/components/datasets/metadata/metadata-dataset/create-metadata-modal.tsx
@@ -36,7 +36,7 @@ const CreateMetadataModal: FC<Props> = ({
         {trigger}
       </PortalToFollowElemTrigger>
       <PortalToFollowElemContent className='z-[1000]'>
-        <CreateContent {...createContentProps} onClose={() => setOpen(false)} />
+        <CreateContent {...createContentProps} onClose={() => setOpen(false)} onBack={() => setOpen(false)} />
       </PortalToFollowElemContent>
     </PortalToFollowElem >
 
diff --git a/web/app/components/datasets/metadata/metadata-dataset/select-metadata-modal.tsx b/web/app/components/datasets/metadata/metadata-dataset/select-metadata-modal.tsx
index ccd0284210..91c97156f4 100644
--- a/web/app/components/datasets/metadata/metadata-dataset/select-metadata-modal.tsx
+++ b/web/app/components/datasets/metadata/metadata-dataset/select-metadata-modal.tsx
@@ -71,6 +71,7 @@ const SelectMetadataModal: FC<Props> = ({
             onSave={handleSave}
             hasBack
             onBack={() => setStep(Step.select)}
+            onClose={() => setStep(Step.select)}
           />
         )}
       </PortalToFollowElemContent>