diff --git a/lib/pages/spaces_management/assign_tag_models/views/assign_tag_models_dialog.dart b/lib/pages/spaces_management/assign_tag_models/views/assign_tag_models_dialog.dart
index 31dba875..90843aa2 100644
--- a/lib/pages/spaces_management/assign_tag_models/views/assign_tag_models_dialog.dart
+++ b/lib/pages/spaces_management/assign_tag_models/views/assign_tag_models_dialog.dart
@@ -15,6 +15,7 @@ import 'package:syncrow_web/pages/spaces_management/space_model/models/tag_model
 import 'package:syncrow_web/pages/spaces_management/space_model/widgets/dialog/create_space_model_dialog.dart';
 import 'package:syncrow_web/pages/spaces_management/tag_model/views/add_device_type_model_widget.dart';
 import 'package:syncrow_web/utils/color_manager.dart';
+import 'package:syncrow_web/pages/spaces_management/helper/tag_helper.dart';
 
 class AssignTagModelsDialog extends StatelessWidget {
   final List<ProductModel>? products;
@@ -258,7 +259,6 @@ class AssignTagModelsDialog extends StatelessWidget {
                                 final processedTags =
                                     result['updatedTags'] as List<TagModel>;
                                 final processedSubspaces = result['subspaces'];
-
                                 if (context.mounted) {
                                   Navigator.of(context).pop();
 
@@ -270,8 +270,10 @@ class AssignTagModelsDialog extends StatelessWidget {
                                             products: products,
                                             subspaces: subspaces,
                                             isCreate: false,
-                                            initialSelectedProducts:
-                                                addedProducts,
+                                            initialSelectedProducts: TagHelper
+                                                .createInitialSelectedProducts(
+                                                    processedTags,
+                                                    processedSubspaces),
                                             allTags: allTags,
                                             spaceName: spaceName,
                                             otherSpaceModels: otherSpaceModels,