diff --git a/mobile/lib/models/gallery_type.dart b/mobile/lib/models/gallery_type.dart
index b73c0e080f..fb07f23a0a 100644
--- a/mobile/lib/models/gallery_type.dart
+++ b/mobile/lib/models/gallery_type.dart
@@ -82,13 +82,13 @@ extension GalleyTypeExtension on GalleryType {
       case GalleryType.locationTag:
       case GalleryType.quickLink:
       case GalleryType.peopleTag:
+      case GalleryType.cluster:
         return true;
       case GalleryType.trash:
       case GalleryType.archive:
       case GalleryType.hiddenSection:
       case GalleryType.hiddenOwnedCollection:
       case GalleryType.sharedCollection:
-      case GalleryType.cluster:
         return false;
     }
   }
diff --git a/mobile/lib/services/machine_learning/face_ml/face_clustering/face_clustering_service.dart b/mobile/lib/services/machine_learning/face_ml/face_clustering/face_clustering_service.dart
index 1319c65409..f3064dbb60 100644
--- a/mobile/lib/services/machine_learning/face_ml/face_clustering/face_clustering_service.dart
+++ b/mobile/lib/services/machine_learning/face_ml/face_clustering/face_clustering_service.dart
@@ -263,8 +263,8 @@ class FaceClusteringService {
   Future<ClusteringResult> predictWithinClusterComputer(
     Map<String, Uint8List> input, {
     Map<int, int>? fileIDToCreationTime,
-    Map<int, (Uint8List, int)> oldClusterSummaries =
-        const <int, (Uint8List, int)>{},
+    Map<String, (Uint8List, int)> oldClusterSummaries =
+        const <String, (Uint8List, int)>{},
     double distanceThreshold = kRecommendedDistanceThreshold,
   }) async {
     _logger.info(
@@ -366,7 +366,7 @@ class FaceClusteringService {
   Future<ClusteringResult> predictCompleteComputer(
     Map<String, Uint8List> input, {
     Map<int, int>? fileIDToCreationTime,
-    required Map<int, (Uint8List, int)> oldClusterSummaries,
+    required Map<String, (Uint8List, int)> oldClusterSummaries,
     double distanceThreshold = kRecommendedDistanceThreshold,
     double mergeThreshold = 0.30,
   }) async {