diff --git a/internal/websocket/broadcast_projection_export_test.go b/internal/websocket/broadcast_projection_export_test.go
index 9869b296d6..f3be2ce1da 100644
--- a/internal/websocket/broadcast_projection_export_test.go
+++ b/internal/websocket/broadcast_projection_export_test.go
@@ -50,6 +50,9 @@ func NewQuietBroadcastProjectionProbeForTest(getState func(string) interface{},
 		hub.dispatchStateBroadcast(&Message{Type: "rawData", Data: stateBroadcastRequest{}}, "")
 		bytes := 0
 		for _, client := range clients {
+			if !hub.clients[client] {
+				return 0, fmt.Errorf("cost probe lost a client")
+			}
 			select {
 			case data := <-client.send:
 				bytes += len(data)
diff --git a/internal/websocket/connected_cost_test.go b/internal/websocket/connected_cost_test.go
index f024a4d676..35eacede49 100644
--- a/internal/websocket/connected_cost_test.go
+++ b/internal/websocket/connected_cost_test.go
@@ -85,6 +85,9 @@ func BenchmarkConnectedDashboardCost(b *testing.B) {
 					if err != nil {
 						b.Fatal(err)
 					}
+					if viewers > 0 && mode != "quiet" && mode != "cold" && wire == 0 {
+						b.Fatal("live workload queued no frame")
+					}
 				}
 				b.ReportMetric(float64(wire), "wire-B/op")
 			})
diff --git a/internal/websocket/delta_dispatch_test.go b/internal/websocket/delta_dispatch_test.go
index bab482c409..706df5b5b1 100644
--- a/internal/websocket/delta_dispatch_test.go
+++ b/internal/websocket/delta_dispatch_test.go
@@ -4,8 +4,6 @@ import (
 	"bytes"
 	"encoding/json"
 	"testing"
-
-	"github.com/rcourtman/pulse-go-rewrite/internal/models"
 )
 
 func TestStateDeltaDispatchSharesOnlyAcceptedBaseline(t *testing.T) {