diff --git a/source/agora/flash/Channel.d b/source/agora/flash/Channel.d
index e48dbb8640b..ef236c667d4 100644
--- a/source/agora/flash/Channel.d
+++ b/source/agora/flash/Channel.d
@@ -739,10 +739,6 @@ LOuter: while (1)
 
     public void onUpdateTxExternalized (in Transaction tx, in uint utxo_idx, in bool is_last)
     {
-        this.state = ChannelState.StartedUnilateralClose;
-        this.onChannelNotify(this.own_pk, this.conf.chan_id, this.state,
-            ErrorCode.None);
-
         this.last_externalized_update_utxo = UTXO.getHash(tx.hashFull(), utxo_idx);
         if (is_last)
         {
@@ -753,7 +749,13 @@ LOuter: while (1)
         }
         else
         {
-            this.publishUpdateTx(this.channel_updates[$ - 1]);
+            this.taskman.setTimer(100.msecs,
+            {
+                this.publishUpdateTx(this.channel_updates[$ - 1]);
+                this.state = ChannelState.StartedUnilateralClose;
+                this.onChannelNotify(this.own_pk, this.conf.chan_id, this.state,
+                    ErrorCode.None, this.height);
+            });
         }
     }
 
@@ -2221,6 +2223,8 @@ LOuter: while (1)
 
     protected Transaction publishUpdateTx (in UpdatePair update)
     {
+        import agora.utils.PrettyPrinter;
+
         auto update_tx = update.update_tx.serializeFull.deserializeFull!Transaction();
         assert(update_tx.inputs.length == 1);
         assert(update_tx.outputs.length == 1);
@@ -2264,9 +2268,11 @@ LOuter: while (1)
                 update_tx.inputs[idx].unlock = genKeyUnlock(fee_sig);
 
         auto result = this.txPublisher(update_tx);
-        log.info("{}: Publishing update tx {}: {}. Result: {}",
+        // NOTE: Added the content of the update tx in the log for debugging
+        // about the failure on the CI build but will be removed after fixing it.
+        log.info("{}: Publishing update tx {}: {}. Result: {}. Tx: {}",
             this.own_pk.flashPrettify, update.seq_id, update_tx.hashFull().flashPrettify,
-            result);
+            result, update_tx.prettify);
         return update_tx;
     }
 
diff --git a/source/agora/flash/Node.d b/source/agora/flash/Node.d
index afc45021b30..81d3f2c974e 100644
--- a/source/agora/flash/Node.d
+++ b/source/agora/flash/Node.d
@@ -730,7 +730,7 @@ public class FlashNode : FlashControlAPI
         if (state == ChannelState.Open)  // todo: might not exist
             this.onChannelOpen(reg_pk, this.channels[chan_id].conf, height);
 
-        this.listener.onChannelNotify(reg_pk, chan_id, state, error);
+        this.listener.onChannelNotify(reg_pk, chan_id, state, error, height);
     }
 
     ///
diff --git a/source/agora/test/Flash.d b/source/agora/test/Flash.d
index 4ee72960d94..1c0cb849569 100644
--- a/source/agora/test/Flash.d
+++ b/source/agora/test/Flash.d
@@ -74,9 +74,10 @@ public interface TestFlashListenerAPI : FlashListenerAPI
     /// has succeeded / failed, and return true / false for success
     ErrorCode waitUntilNotified (Invoice);
 
-    /// wait until we get a notification about the given channel state,
-    /// and return any associated error codes
-    ErrorCode waitUntilChannelState (Hash, ChannelState, PublicKey node = PublicKey.init);
+    /// wait until we get a notification about the given channel state
+    /// with the specified height and return any associated error codes
+    ErrorCode waitUntilChannelState (Hash, ChannelState, PublicKey node = PublicKey.init,
+        Height height = Height(0), bool erase_state = true);
 
     /// Print out the contents of the log
     public void printLog ();
@@ -498,6 +499,7 @@ private class FlashListener : TestFlashListenerAPI
     {
         ChannelState state;
         ErrorCode error;
+        Height height;
     }
 
     State[PublicKey][Hash] channel_state;
@@ -541,16 +543,24 @@ private class FlashListener : TestFlashListenerAPI
     }
 
     public ErrorCode waitUntilChannelState (Hash chan_id, ChannelState state,
-        PublicKey node = PublicKey.init)
+        PublicKey node = PublicKey.init, Height height = Height(0),
+        bool erase_state = true)
     {
-        scope (exit) this.channel_state.remove(chan_id);
+        // There is no case where we check the state with a specified
+        // height and no public key.
+        assert(height == Height(0) || node != PublicKey.init);
+
+        scope (exit) if (erase_state) this.channel_state.remove(chan_id);
         while (1)
         {
             if (auto chan_states = chan_id in this.channel_state)
             {
                 if (auto chan_state = node in *chan_states)
-                    if ((*chan_state).state >= state)
+                {
+                    if ((*chan_state).state >= state &&
+                        (*chan_state).height >= height)
                         return (*chan_state).error;
+                }
                 if (node == PublicKey.init)
                 {
                     auto states = chan_states.byValue
@@ -565,12 +575,12 @@ private class FlashListener : TestFlashListenerAPI
     }
 
     public void onChannelNotify (PublicKey pk, Hash chan_id, ChannelState state,
-        ErrorCode error, Height = Height(0))
+        ErrorCode error, Height height = Height(0))
     {
         log.info("Channel event {}, id {}", state, chan_id.flashPrettify);
         if (chan_id !in this.channel_state)
             this.channel_state[chan_id] = typeof(this.channel_state[chan_id]).init;
-        this.channel_state[chan_id][pk] = State(state, error);
+        this.channel_state[chan_id][pk] = State(state, error, height);
     }
 
     public Result!ChannelUpdate onRequestedChannelOpen (PublicKey pk, ChannelConfig conf)
@@ -1924,6 +1934,13 @@ unittest
     network.clients[0].postTransaction(update_tx);
     assert(!network.clients[0].hasTransactionHash(update_tx.hashFull()));
 
+    // wait until all the flash nodes skip publishing `update tx` before
+    // the `setPublishEnable` is called.
+    network.listener.waitUntilChannelState(chan_id,
+        ChannelState.StartedUnilateralClose, WK.Keys.A.address, Height(4), false);
+    network.listener.waitUntilChannelState(chan_id,
+        ChannelState.StartedUnilateralClose, WK.Keys.C.address, Height(4));
+
     // allow normal node operation again
     alice.setPublishEnable(true);
     charlie.setPublishEnable(true);