From f118b21a30b134034ec0559579c1b6d0ef535ebe Mon Sep 17 00:00:00 2001
From: waterplea <alexander@inkin.ru>
Date: Fri, 5 Aug 2022 21:30:59 +0300
Subject: [PATCH] chore: small tweaks

---
 client/lib/source/ws-source.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/client/lib/source/ws-source.ts b/client/lib/source/ws-source.ts
index 3470cd0..fedb739 100644
--- a/client/lib/source/ws-source.ts
+++ b/client/lib/source/ws-source.ts
@@ -1,10 +1,10 @@
 import { Observable } from 'rxjs'
-import { webSocket, WebSocketSubject } from 'rxjs/webSocket'
+import { webSocket } from 'rxjs/webSocket'
 import { Update } from '../types'
 import { Source } from './source'
 
 export class WebsocketSource<T> implements Source<T> {
-  private websocket$: WebSocketSubject<RPCResponse<Update<T>>> = webSocket({
+  private websocket$ = webSocket<RPCResponse<Update<T>>>({
     url: this.url,
     openObserver: {
       next: () => {
@@ -15,7 +15,8 @@ export class WebsocketSource<T> implements Source<T> {
 
   constructor(
     private readonly url: string,
-    private readonly document: Document,
+    // TODO: Remove fallback after client app is updated
+    private readonly document: Document = document,
   ) {}
 
   watch$(): Observable<RPCResponse<Update<T>>> {