diff --git a/test/features/create_diet/integration_widget_test.dart b/test/features/create_diet/integration_widget_test.dart
index 0d8c7b31..541056f6 100644
--- a/test/features/create_diet/integration_widget_test.dart
+++ b/test/features/create_diet/integration_widget_test.dart
@@ -326,7 +326,7 @@ void main() {
       expect(find.byType(DietTypeChoiceWidget), findsOneWidget);
       expect(find.byType(DietImagePickerWidget), findsOneWidget);
       // Action buttons
-      expect(find.byType(CustomOutlinedButton), findsNWidgets(3));
+      expect(find.byType(CustomOutlinedButton), findsNWidgets(2));
     });
   });
 }
diff --git a/test/features/my_diets/presentation/widgets/basic_widget_test.dart b/test/features/my_diets/presentation/widgets/basic_widget_test.dart
index 9afb6818..ef674d0f 100644
--- a/test/features/my_diets/presentation/widgets/basic_widget_test.dart
+++ b/test/features/my_diets/presentation/widgets/basic_widget_test.dart
@@ -106,7 +106,7 @@ void main() {
 
       // Assert
       expect(find.byType(MyDietsTab), findsOneWidget);
-      expect(find.byType(AnimatedOptionButton), findsNWidgets(3));
+      expect(find.byType(AnimatedOptionButton), findsNWidgets(2));
     });
 
     testWidgets('should handle tab selection in MyDietsTab', (WidgetTester tester) async {
@@ -119,7 +119,7 @@ void main() {
 
       // Act - Find and tap the second tab
       final tabButtons = find.byType(AnimatedOptionButton);
-      expect(tabButtons, findsNWidgets(3));
+      expect(tabButtons, findsNWidgets(2));
 
       // Tap the second tab (index 1)
       await tester.tap(tabButtons.at(1));
@@ -221,7 +221,7 @@ void main() {
 
       // Assert
       expect(find.byType(MyDietsTab), findsOneWidget);
-      expect(find.byType(AnimatedOptionButton), findsNWidgets(3));
+      expect(find.byType(AnimatedOptionButton), findsNWidgets(2));
     });
 
     testWidgets('should handle MyDietsTab additionalOnTap callback', (WidgetTester tester) async {