for (i = 0; i < isp->num_of_streams; i++) {
                struct atomisp_sub_device *asd = &isp->asd[i];
-#ifndef ISP2401
-               timer_setup(&asd->wdt, atomisp_wdt, 0);
-#else
-               timer_setup(&asd->video_out_capture.wdt, atomisp_wdt, 0);
-               timer_setup(&asd->video_out_preview.wdt, atomisp_wdt, 0);
-               timer_setup(&asd->video_out_vf.wdt, atomisp_wdt, 0);
-               timer_setup(&asd->video_out_video_capture.wdt, atomisp_wdt, 0);
-#endif
+               if (!atomisp_hw_is_isp2401)
+                       timer_setup(&asd->wdt, atomisp_wdt, 0);
+               else {
+                       timer_setup(&asd->video_out_capture.wdt,
+                                   atomisp_wdt, 0);
+                       timer_setup(&asd->video_out_preview.wdt,
+                                   atomisp_wdt, 0);
+                       timer_setup(&asd->video_out_vf.wdt, atomisp_wdt, 0);
+                       timer_setup(&asd->video_out_video_capture.wdt,
+                                   atomisp_wdt, 0);
+               }
        }
        return 0;
 alloc_fail: