diff --git a/flutter/lib/features/import/presentation/receipt_import_tab.dart b/flutter/lib/features/import/presentation/receipt_import_tab.dart
index 705f270e..2b5ed286 100644
--- a/flutter/lib/features/import/presentation/receipt_import_tab.dart
+++ b/flutter/lib/features/import/presentation/receipt_import_tab.dart
@@ -349,6 +349,7 @@ class _EditDialogState extends State<_EditDialog> {
 
   void _applyAiSuggestionForExistingSelection() {
     final suggestedId = widget.item.suggestedProductId;
+    int? preselectedCategoryId = widget.item.categorySuggestionId;
     if (suggestedId != null) {
       setState(() {
         _productId = suggestedId;
@@ -360,11 +361,11 @@ class _EditDialogState extends State<_EditDialog> {
             _categoryPathForCategoryId(_productCategoryId) ?? widget.item.categorySuggestionPath;
         _productCategorySource = CategorySelectionSource.ai;
       });
-      return;
+      preselectedCategoryId = _productCategoryId;
     }
 
     _openExistingCategoryPicker(
-      preselectedCategoryId: widget.item.categorySuggestionId,
+      preselectedCategoryId: preselectedCategoryId,
     );
   }
 
@@ -634,7 +635,9 @@ class _EditDialogState extends State<_EditDialog> {
                       backgroundColor: Colors.green.shade50,
                       side: BorderSide(color: Colors.green.shade300),
                       visualDensity: VisualDensity.compact,
-                      onPressed: _applyAiSuggestionForExistingSelection,
+                      onPressed: () => _openExistingCategoryPicker(
+                        preselectedCategoryId: item.categorySuggestionId,
+                      ),
                     ),
                   ],
                 ],