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) }