diff --git a/src/components/ui/Transition.tsx b/src/components/ui/Transition.tsx
index 257eb6901..4cfa408ca 100644
--- a/src/components/ui/Transition.tsx
+++ b/src/components/ui/Transition.tsx
@@ -226,6 +226,7 @@ const Transition: FC<OwnProps & StateProps> = ({
       if (activeElement) {
         activeElement.style.height = 'auto';
         container.style.height = `${activeElement.clientHeight}px`;
+        container.style.flexBasis = `${activeElement.clientHeight}px`;
       }
     }
   }, [shouldRestoreHeight, children]);