dev_dbg(isp->dev, "atomisp css flush done\n");
 }
 
-#ifndef ISP2401
-void atomisp_wdt(unsigned long isp_addr)
-#else
-void atomisp_wdt(unsigned long pipe_addr)
-#endif
+void atomisp_wdt(struct timer_list *t)
 {
 #ifndef ISP2401
-       struct atomisp_device *isp = (struct atomisp_device *)isp_addr;
+       struct atomisp_sub_device *asd = from_timer(asd, t, wdt);
 #else
-       struct atomisp_video_pipe *pipe =
-               (struct atomisp_video_pipe *)pipe_addr;
+       struct atomisp_video_pipe *pipe = from_timer(pipe, t, wdt);
        struct atomisp_sub_device *asd = pipe->asd;
-       struct atomisp_device *isp = asd->isp;
 #endif
+       struct atomisp_device *isp = asd->isp;
 
 #ifdef ISP2401
        atomic_inc(&pipe->wdt_count);
 
 void atomisp_msi_irq_init(struct atomisp_device *isp, struct pci_dev *dev);
 void atomisp_msi_irq_uninit(struct atomisp_device *isp, struct pci_dev *dev);
 void atomisp_wdt_work(struct work_struct *work);
-#ifndef ISP2401
-void atomisp_wdt(unsigned long isp_addr);
-#else
-void atomisp_wdt(unsigned long pipe_addr);
-#endif
+void atomisp_wdt(struct timer_list *t);
 void atomisp_setup_flash(struct atomisp_sub_device *asd);
 irqreturn_t atomisp_isr(int irq, void *dev);
 irqreturn_t atomisp_isr_thread(int irq, void *isp_ptr);
 
                        for (i = 0; i < isp->num_of_streams; i++)
                                atomisp_wdt_stop(&isp->asd[i], 0);
 #ifndef ISP2401
-                       atomisp_wdt((unsigned long)isp);
+                       atomisp_wdt(&isp->asd[0].wdt);
 #else
                        queue_work(isp->wdt_work_queue, &isp->wdt_work);
 #endif
 
                struct atomisp_sub_device *asd = &isp->asd[i];
                asd = &isp->asd[i];
 #ifndef ISP2401
-               setup_timer(&asd->wdt, atomisp_wdt, (unsigned long)isp);
+               timer_setup(&asd->wdt, atomisp_wdt, 0);
 #else
-               setup_timer(&asd->video_out_capture.wdt,
-                       atomisp_wdt, (unsigned long)&asd->video_out_capture);
-               setup_timer(&asd->video_out_preview.wdt,
-                       atomisp_wdt, (unsigned long)&asd->video_out_preview);
-               setup_timer(&asd->video_out_vf.wdt,
-                       atomisp_wdt, (unsigned long)&asd->video_out_vf);
-               setup_timer(&asd->video_out_video_capture.wdt,
-                       atomisp_wdt,
-                       (unsigned long)&asd->video_out_video_capture);
+               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
        }
        return 0;