AppUpdateManager concurrent modification crash

ListenerTest.kt

@Test
@LooperMode(LooperMode.Mode.PAUSED)
fun crashesWhenUnsubscribing() {
    updateManager.setUpdateAvailable(100500)
    updateManager.withInfo {
        startUpdateFlowForResult(it, AppUpdateType.FLEXIBLE, activity, 100)
        assertTrue(isConfirmationDialogVisible)
        userAcceptsUpdate()
        downloadStarts()

        (1..2).map {
            registerListener(object: InstallStateUpdatedListener {
                override fun onStateUpdate(p0: InstallState?) {
                    unregisterListener(this)
                }
            })
        }

        assertFailsWith<ConcurrentModificationException> {
            downloadCompletes()
            shadowOf(getMainLooper()).idle()
        }
    }
    shadowOf(getMainLooper()).idle()
}