spin_lock_bh(&ioat_chan->prep_lock);
                check_active(ioat_chan);
                spin_unlock_bh(&ioat_chan->prep_lock);
-               spin_unlock_bh(&ioat_chan->cleanup_lock);
-               return;
+               goto unlock_out;
+       }
+
+       /* handle the missed cleanup case */
+       if (ioat_cleanup_preamble(ioat_chan, &phys_complete)) {
+               /* timer restarted in ioat_cleanup_preamble
+                * and IOAT_COMPLETION_ACK cleared
+                */
+               __cleanup(ioat_chan, phys_complete);
+               goto unlock_out;
        }
 
        /* if we haven't made progress and we have already
         * acknowledged a pending completion once, then be more
         * forceful with a restart
         */
-       if (ioat_cleanup_preamble(ioat_chan, &phys_complete))
-               __cleanup(ioat_chan, phys_complete);
-       else if (test_bit(IOAT_COMPLETION_ACK, &ioat_chan->state)) {
+       if (test_bit(IOAT_COMPLETION_ACK, &ioat_chan->state)) {
                u32 chanerr;
 
                chanerr = readl(ioat_chan->reg_base + IOAT_CHANERR_OFFSET);
                        ioat_ring_active(ioat_chan));
 
                ioat_reboot_chan(ioat_chan);
-               spin_unlock_bh(&ioat_chan->cleanup_lock);
-               return;
-       } else
-               set_bit(IOAT_COMPLETION_ACK, &ioat_chan->state);
 
+               goto unlock_out;
+       }
+
+       set_bit(IOAT_COMPLETION_ACK, &ioat_chan->state);
        mod_timer(&ioat_chan->timer, jiffies + COMPLETION_TIMEOUT);
+unlock_out:
        spin_unlock_bh(&ioat_chan->cleanup_lock);
 }