@mitchwongho: As @dalexander correctly pointed out, the main issue is in the type of the map passed to view model. You need Map<String, (State, Any) -> State> but inferred type is Map<String, (State, Nothing) -> State>