Reimplement flush_kthread work