Android Studio AlertDialog Pass Values to DialogFragment - Kotlin 2

MyDialogFragment.kt

class MyDialogFragment : DialogFragment() {

    override fun onCreateDialog(savedInstanceState: Bundle?): Dialog {
        val dialog = activity?.let {
            // 値を受け取る
            val title = arguments?.getString("TITLE", "")
            val number = arguments?.getInt("NUMBER", 0)

            AlertDialog.Builder(it)
                .setTitle(title)
                .setMessage("受け取った値は$number")
                .setPositiveButton("OK") { _, _ -> }
                .create()
        }
        return dialog ?: throw IllegalStateException("アクティビティがNullです。")
    }
}