Update

web/apps/photos/src/components/Collections/AllAlbums.tsx

@@ -20,7 +20,7 @@ import {
 } from "ente-new/photos/components/Tiles";
 import type { CollectionSummary } from "ente-new/photos/services/collection/ui";
 import { CollectionsSortBy } from "ente-new/photos/services/collection/ui";
-import { FlexWrapper, FluidContainer } from "ente-shared/components/Container";
+import { FlexWrapper } from "ente-shared/components/Container";
 import { t } from "i18next";
 import memoize from "memoize-one";
 import React, { useEffect, useRef, useState } from "react";
@@ -104,8 +104,8 @@ const Title = ({
     isInHiddenSection,
 }) => (
     <DialogTitle>
-        <FlexWrapper>
-            <FluidContainer mr={1.5}>
+        <Stack direction="row" sx={{ gap: 1.5 }}>
+            <Stack sx={{ flex: 1 }}>
                 <Box>
                     <Typography variant="h5">
                         {isInHiddenSection
@@ -123,18 +123,16 @@ const Title = ({
                         {t("albums_count", { count: collectionCount })}
                     </Typography>
                 </Box>
-            </FluidContainer>
-            <Stack direction="row" sx={{ gap: 1.5 }}>
-                <CollectionsSortOptions
-                    activeSortBy={collectionsSortBy}
-                    onChangeSortBy={onChangeCollectionsSortBy}
-                    nestedInDialog
-                />
-                <FilledIconButton onClick={onClose}>
-                    <CloseIcon />
-                </FilledIconButton>
             </Stack>
-        </FlexWrapper>
+            <CollectionsSortOptions
+                activeSortBy={collectionsSortBy}
+                onChangeSortBy={onChangeCollectionsSortBy}
+                nestedInDialog
+            />
+            <FilledIconButton onClick={onClose}>
+                <CloseIcon />
+            </FilledIconButton>
+        </Stack>
     </DialogTitle>
 );