foo.diff

diff --git a/src/protocol/socket.ts b/src/protocol/socket.ts
index ae5d602c..044b3f10 100644
--- a/src/protocol/socket.ts
+++ b/src/protocol/socket.ts
@@ -1,5 +1,6 @@
 import { defer, makeInfallible } from "./utils";
 import { ProtocolClient } from "@recordreplay/protocol";
+import { setExpectedError } from "ui/actions/session";
 
 interface Message {
   id: number;
@@ -28,12 +29,12 @@ const gStartTime = Date.now();
 let gSentBytes = 0;
 let gReceivedBytes = 0;
 
-export function initSocket(address?: string) {
+export function initSocket(store, address?: string) {
   socket = new WebSocket(address || "wss://dispatch.replay.io");
 
   socket.onopen = makeInfallible(onSocketOpen);
   socket.onclose = makeInfallible(onSocketClose);
-  socket.onerror = makeInfallible(onSocketError);
+  socket.onerror = makeInfallible(() => store.dispatch(onSocketError()));
   socket.onmessage = makeInfallible(onSocketMessage);
 }
 
@@ -116,7 +117,10 @@ function onSocketClose() {
 }
 
 function onSocketError() {
-  log("Socket Error");
+  return ({ dispatch }) => {
+    log("Socket Error");
+    dispatch(setExpectedError);
+  };
 }
 
 export function log(text: string) {