* Callback used when waiting on sets with l_wait_event.
  * Always returns 1.
  */
-int ptlrpc_expired_set(void *data)
+void ptlrpc_expired_set(struct ptlrpc_request_set *set)
 {
-       struct ptlrpc_request_set *set = data;
        struct list_head *tmp;
        time64_t now = ktime_get_real_seconds();
 
                 */
                ptlrpc_expire_one_request(req, 1);
        }
+}
+static int ptlrpc_expired_set_void(void *data)
+{
+       struct ptlrpc_request_set *set = data;
 
+       ptlrpc_expired_set(set);
        /*
         * When waiting for a whole set, we always break out of the
         * sleep so we can recalculate the timeout, or enable interrupts
                         * so we allow interrupts during the timeout.
                         */
                        lwi = LWI_TIMEOUT_INTR_ALL(HZ,
-                                                  ptlrpc_expired_set,
+                                                  ptlrpc_expired_set_void,
                                                   ptlrpc_interrupted_set, set);
                else
                        /*
                         * complete, or an in-flight req times out.
                         */
                        lwi = LWI_TIMEOUT((timeout ? timeout : 1) * HZ,
-                                         ptlrpc_expired_set, set);
+                                         ptlrpc_expired_set_void, set);
 
                rc = l_wait_event(set->set_waitq, ptlrpc_check_set(NULL, set), &lwi);
 
 
 void ptlrpc_init_xid(void);
 void ptlrpc_set_add_new_req(struct ptlrpcd_ctl *pc,
                            struct ptlrpc_request *req);
-int ptlrpc_expired_set(void *data);
+void ptlrpc_expired_set(struct ptlrpc_request_set *set);
 int ptlrpc_set_next_timeout(struct ptlrpc_request_set *set);
 void ptlrpc_resend_req(struct ptlrpc_request *request);
 void ptlrpc_set_bulk_mbits(struct ptlrpc_request *req);
 
         * new_req_list and ptlrpcd_check() moves them into the set.
         */
        do {
-               struct l_wait_info lwi;
                int timeout;
 
                timeout = ptlrpc_set_next_timeout(set);
-               lwi = LWI_TIMEOUT((timeout ? timeout : 1) * HZ,
-                                 ptlrpc_expired_set, set);
 
                lu_context_enter(&env.le_ctx);
                lu_context_enter(env.le_ses);
-               l_wait_event(set->set_waitq, ptlrpcd_check(&env, pc), &lwi);
+               if (wait_event_idle_timeout(set->set_waitq,
+                                           ptlrpcd_check(&env, pc),
+                                           (timeout ? timeout : 1) * HZ) == 0)
+                       ptlrpc_expired_set(set);
+
                lu_context_exit(&env.le_ctx);
                lu_context_exit(env.le_ses);