diff --git a/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.graphql b/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.graphql
index 97175f1fa7..488f951260 100644
--- a/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.graphql
+++ b/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.graphql
@@ -5,9 +5,7 @@ mutation CreateMedia($input: UploadProgressReportMedia!) {
       items {
         id
         variant {
-          key
-          label
-          responsibleRole
+          ...variant
         }
         category
       }
diff --git a/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.tsx b/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.tsx
index 65e850ddbc..240782db57 100644
--- a/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.tsx
+++ b/src/scenes/ProgressReports/EditForm/Steps/Media/MediaStep.tsx
@@ -66,20 +66,18 @@ export const MediaStep: StepComponent = ({ report }) => {
                   {...getInputProps()}
                   name={`upload-media-${variant.key}`}
                 />
-                <Box sx={{ display: 'block' }} key={variant.key}>
-                  <VariantMediaAccordion
-                    deleteMedia={deleteMedia}
-                    openFileBrowser={openFileBrowser}
-                    response={{
-                      item: report.media.items.filter(
-                        (item) => item.variant.key === variant.key
-                      ),
-                      uploadVariant: variant,
-                    }}
-                    sensitivity={report.parent.sensitivity}
-                    initiallyOpen={false}
-                  />
-                </Box>
+                <VariantMediaAccordion
+                  deleteMedia={deleteMedia}
+                  openFileBrowser={openFileBrowser}
+                  response={{
+                    item: report.media.items.filter(
+                      (item) => item.variant.key === variant.key
+                    ),
+                    uploadVariant: variant,
+                  }}
+                  sensitivity={report.parent.sensitivity}
+                  initiallyOpen={false}
+                />
               </Fragment>
             );
           })}
diff --git a/src/scenes/ProgressReports/EditForm/Steps/Media/VariantMediaAccordion.tsx b/src/scenes/ProgressReports/EditForm/Steps/Media/VariantMediaAccordion.tsx
index a7a5dc8a04..20dcb8bb73 100644
--- a/src/scenes/ProgressReports/EditForm/Steps/Media/VariantMediaAccordion.tsx
+++ b/src/scenes/ProgressReports/EditForm/Steps/Media/VariantMediaAccordion.tsx
@@ -265,23 +265,21 @@ export const VariantMediaAccordion = ({
         <RoleIcon variantRole={response.uploadVariant.responsibleRole} />
         <span>{response.uploadVariant.label}</span>
       </AccordionSummary>
-      <AccordionDetails>
-        <Box sx={{ display: 'flex', p: '0px 16px' }}>
-          <ImageBox
-            media={response.item}
-            openFileBrowser={openFileBrowser}
-            deleteMedia={deleteCurrentMedia}
-            sensitivity={sensitivity}
-          />
-          <MediaInfoForm
-            initialValues={{
-              category: response.item[0]?.category,
-              caption: response.item[0]?.media?.caption,
-              id: response.item[0]?.id,
-            }}
-            disabled={!response.item}
-          />
-        </Box>
+      <AccordionDetails sx={{ display: 'flex', p: '0px 16px' }}>
+        <ImageBox
+          media={response.item}
+          openFileBrowser={openFileBrowser}
+          deleteMedia={deleteCurrentMedia}
+          sensitivity={sensitivity}
+        />
+        <MediaInfoForm
+          initialValues={{
+            category: response.item[0]?.category,
+            caption: response.item[0]?.media?.caption,
+            id: response.item[0]?.id,
+          }}
+          disabled={!response.item}
+        />
       </AccordionDetails>
     </Accordion>
   );