From 1b59f2257df569ad3d47f36ed3c028c8d0a3bbca Mon Sep 17 00:00:00 2001
From: Kirill Nayduik <kirill.nayduikkn1@gmail.com>
Date: Fri, 30 Jul 2021 15:00:47 +0300
Subject: [PATCH] Fix PR comment:
 https://github.com/TouchInstinct/RoboSwag/pull/204#discussion_r662247564

---
 .../java/ru/touchin/roboswag/mvi_arch/core/Extensions.kt  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/mvi-arch/src/main/java/ru/touchin/roboswag/mvi_arch/core/Extensions.kt b/mvi-arch/src/main/java/ru/touchin/roboswag/mvi_arch/core/Extensions.kt
index ae46f6e..4e6c320 100644
--- a/mvi-arch/src/main/java/ru/touchin/roboswag/mvi_arch/core/Extensions.kt
+++ b/mvi-arch/src/main/java/ru/touchin/roboswag/mvi_arch/core/Extensions.kt
@@ -1,9 +1,15 @@
 package ru.touchin.roboswag.mvi_arch.core
 
 import android.os.Parcelable
+import ru.touchin.roboswag.mvi_arch.marker.ViewAction
+import ru.touchin.roboswag.mvi_arch.marker.ViewState
 
 /**
  * Used for setting arguments and initial state into Fragments
  */
-fun <NavArgs : Parcelable, State, Action, VM, TFragment : MviFragment<NavArgs, State, Action, VM>>
+fun <NavArgs : Parcelable,
+        State : ViewState,
+        Action : ViewAction,
+        VM : MviViewModel<NavArgs, Action, State>,
+        TFragment : MviFragment<NavArgs, State, Action, VM>>
         TFragment.withArgs(navArgs: NavArgs) = apply { initArgs(navArgs) }