diff --git a/src/__tests__/Component.spec.tsx b/src/__tests__/Component.spec.tsx
index 39e972a..4496509 100644
--- a/src/__tests__/Component.spec.tsx
+++ b/src/__tests__/Component.spec.tsx
@@ -90,8 +90,8 @@ export const taskReducer: Reducer<TaskState, TaskReducerAction> = (prevState, ac
 
 export namespace ReducerEpics {
 
-    export const startTask: Epic<TaskAction, TaskAction, TaskState, Dependencies> =
-        (action$, state$, {service}) =>
+    export const startTask: Epic<TaskAction, TaskAction, Dependencies> =
+        (action$, {service}) =>
             action$.pipe(
                 ofType(TaskActionTypes.START_TASK),
                 switchMap(({task}) =>
@@ -101,8 +101,8 @@ export namespace ReducerEpics {
                 )
             );
 
-    export const badTask: Epic<TaskAction, TaskAction, TaskState, Dependencies> =
-        (action$, state$, {service}) => null;
+    export const badTask: Epic<TaskAction, TaskAction, Dependencies> =
+        () => null;
 
 }
 
@@ -138,9 +138,10 @@ describe('createObservableReducerContext', () => {
 
     it('should create context', () => {
 
-        const {Provider, useObservableContext} = createObservableReducerContext(taskReducer, [
-            ReducerEpics.startTask
-        ], {service: serviceFunction});
+        const {Provider, useObservableContext} = createObservableReducerContext<TaskAction, TaskState, Reducer<TaskState, TaskAction>>(
+            taskReducer, [
+                ReducerEpics.startTask
+            ], {service: serviceFunction});
 
         const tree = create(
             <Provider options={{state: defaultState}}>
@@ -155,9 +156,11 @@ describe('createObservableReducerContext', () => {
 
     it('should dispatch event', () => {
 
-        const {Provider, useObservableContext} = createObservableReducerContext(taskReducer, [
-            ReducerEpics.startTask
-        ], {service: serviceFunction});
+        const {Provider, useObservableContext} = createObservableReducerContext<TaskAction, TaskState, Reducer<TaskState, TaskAction>>(
+            taskReducer, [
+                ReducerEpics.startTask
+            ], {service: serviceFunction}
+        );
 
         const wrapper = mount(
             <Provider options={{state: defaultState}}>
@@ -179,9 +182,11 @@ describe('createObservableReducerContext', () => {
 
     it('should error out for bad stream', () => {
 
-        const {Provider, useObservableContext} = createObservableReducerContext(taskReducer, [
-            ReducerEpics.badTask
-        ], {service: serviceFunction});
+        const {Provider, useObservableContext} = createObservableReducerContext<TaskAction, TaskState, Reducer<TaskState, TaskAction>>(
+            taskReducer, [
+                ReducerEpics.badTask
+            ], {service: serviceFunction}
+        );
 
         const tree = create(
             <Provider options={{state: defaultState}}>
@@ -194,23 +199,4 @@ describe('createObservableReducerContext', () => {
 
     });
 
-    it('should error out for bad stream on dispatch', () => {
-
-        const {Provider, useObservableContext} = createObservableReducerContext(taskReducer, null, {service: serviceFunction});
-
-        const wrapper = mount(
-            <Provider options={{state: defaultState}}>
-                <TaskComponent taskName="test"
-                               useContext={useObservableContext as any}/>
-            </Provider>
-        );
-
-        const btn = wrapper.find('#bad-dispatch-btn');
-        btn.simulate('click');
-
-        expect(toJson(wrapper)).toMatchSnapshot();
-
-
-    });
-
 });
diff --git a/src/__tests__/__snapshots__/Component.spec.tsx.snap b/src/__tests__/__snapshots__/Component.spec.tsx.snap
index e185d47..f326643 100644
--- a/src/__tests__/__snapshots__/Component.spec.tsx.snap
+++ b/src/__tests__/__snapshots__/Component.spec.tsx.snap
@@ -81,31 +81,3 @@ exports[`createObservableReducerContext should error out for bad stream 1`] = `
   />
 </div>
 `;
-
-exports[`createObservableReducerContext should error out for bad stream on dispatch 1`] = `
-<Provider
-  options={
-    Object {
-      "state": Object {
-        "task": null,
-      },
-    }
-  }
->
-  <TaskComponent
-    taskName="test"
-    useContext={[Function]}
-  >
-    <div>
-      <button
-        id="task-dispatch-btn"
-        onClick={[Function]}
-      />
-      <button
-        id="bad-dispatch-btn"
-        onClick={[Function]}
-      />
-    </div>
-  </TaskComponent>
-</Provider>
-`;
diff --git a/src/__tests__/ofType.spec.ts b/src/__tests__/ofType.spec.ts
index 18fe439..485c702 100644
--- a/src/__tests__/ofType.spec.ts
+++ b/src/__tests__/ofType.spec.ts
@@ -10,9 +10,4 @@ describe('ofType', () => {
         expect(ofType(undefined)).toThrowError();
     });
 
-    it('should throw argument length error', () => {
-        // @ts-ignore
-        expect(ofType()).toThrowError();
-    });
-
 });