Add set_current_state() before schedule_timeout() so that if the
while-loop iterates multiple times, schedule_timeout() delays as
expected. Without the addition, schedule_timeout() will return
immediately.
Signed-off-by: Greg Kroah-Hartman <greg@kroah.com>