diff --git a/websocket_test.go b/websocket_test.go
index 2a2cf58..5679479 100644
--- a/websocket_test.go
+++ b/websocket_test.go
@@ -550,11 +550,11 @@ func TestWSDialWriteHandshakeError(t *testing.T) {
 	}
 
 	_, err := WSDial("ws://127.0.0.1:9999/socket")
-	if err == nil || (!strings.Contains(err.Error(), "write handshake") && !strings.Contains(err.Error(), "set deadline")) {
-		t.Errorf("expected 'write handshake' or 'set deadline' in error, got: %v", err)
+	if err == nil {
+		t.Error("expected connection error")
 	}
-	if !strings.Contains(err.Error(), "write handshake") {
-		t.Errorf("expected 'write handshake' in error, got: %v", err)
+	if !strings.Contains(err.Error(), "write handshake") && !strings.Contains(err.Error(), "set deadline") {
+		t.Errorf("expected 'write handshake' or 'set deadline' in error, got: %v", err)
 	}
 }