diff --git a/notifications/notifications.go b/notifications/notifications.go
index defea700aee48fd68cbaf24c4eacd29da2a37f24..9a6f10b525610a1f65f80d10029243810faa4a2d 100644
--- a/notifications/notifications.go
+++ b/notifications/notifications.go
@@ -36,6 +36,16 @@ type impl struct {
 }
 
 func (ps *impl) Publish(block blocks.Block) {
+	ps.wg.Add(1)
+	defer ps.wg.Done()
+
+	select {
+	case <-ps.cancel:
+		// Already shutdown, bail.
+		return
+	default:
+	}
+
 	ps.wrapped.Pub(block, block.Cid().KeyString())
 }