diff --git a/lib/features/scene/widgets/bottom_sheet_widget.dart b/lib/features/scene/widgets/bottom_sheet_widget.dart
index 872e1eb..887fa27 100644
--- a/lib/features/scene/widgets/bottom_sheet_widget.dart
+++ b/lib/features/scene/widgets/bottom_sheet_widget.dart
@@ -95,7 +95,7 @@ class CustomBottomSheetWidget extends StatelessWidget {
         deviceName: 'Delay The Action',
         icon: '',
         operationName: 'Delay The Action',
-        code: 'delay',
+        code: '',
         functionValue: 0,
         operationalValues: [
           SceneOperationalValue(icon: '', value: 0),
diff --git a/lib/features/scene/widgets/create_scene_save_button.dart b/lib/features/scene/widgets/create_scene_save_button.dart
index 29a0e9f..179153d 100644
--- a/lib/features/scene/widgets/create_scene_save_button.dart
+++ b/lib/features/scene/widgets/create_scene_save_button.dart
@@ -1,9 +1,6 @@
 import 'package:flutter/material.dart';
 import 'package:flutter_bloc/flutter_bloc.dart';
-import 'package:syncrow_app/features/app_layout/bloc/home_cubit.dart';
 import 'package:syncrow_app/features/scene/bloc/create_scene/create_scene_bloc.dart';
-import 'package:syncrow_app/features/scene/bloc/scene_bloc/scene_bloc.dart';
-import 'package:syncrow_app/features/scene/bloc/scene_bloc/scene_event.dart';
 import 'package:syncrow_app/features/scene/helper/scene_logic_helper.dart';
 import 'package:syncrow_app/features/shared_widgets/default_button.dart';
 import 'package:syncrow_app/features/shared_widgets/text_widgets/body_large.dart';
diff --git a/lib/features/scene/widgets/if_then_containers/then_added_tasks.dart b/lib/features/scene/widgets/if_then_containers/then_added_tasks.dart
index f5a99cd..c1870b1 100644
--- a/lib/features/scene/widgets/if_then_containers/then_added_tasks.dart
+++ b/lib/features/scene/widgets/if_then_containers/then_added_tasks.dart
@@ -33,7 +33,8 @@ class ThenAddedTasksContainer extends StatelessWidget
   @override
   Widget build(BuildContext context) {
     String operationValue = '';
-    if (taskItem.code.contains('countdown')) {
+    if (taskItem.code.contains('countdown') ||
+        taskItem.deviceId.contains('delay')) {
       final functionValue =
           taskItem.functionValue ?? taskItem.operationalValues.first.value;
       final duration =