Kotlin way ViewModelFactory with dagger2

ViewModelFactory.kt

import android.arch.lifecycle.ViewModel
import android.arch.lifecycle.ViewModelProvider
import javax.inject.Inject
import javax.inject.Provider
import javax.inject.Singleton

@Singleton
class ViewModelFactory @Inject constructor(
        private val creators: Map<Class<out ViewModel>,
        @JvmSuppressWildcards Provider<ViewModel>>
) : ViewModelProvider.Factory {
    
    override fun <T : ViewModel> create(modelClass: Class<T>): T {
        val creator = creators[modelClass]
                ?: creators.entries.firstOrNull { (key, _) ->
                    modelClass.isAssignableFrom(key)
                }?.value
                ?: throw IllegalArgumentException("unknown model class " + modelClass)

        return creator.get() as T
    }
}