Consegui que funcionasse com as seguintes alterações:
Para o comutador sendo substituído, conecte o fio preto a BK, o outro fio preto a RD e os dois fios brancos que foram tampados juntos agora estão conectados a WH no novo comutador.
Para a chave que está sendo removida, deixamos dois fios brancos juntos e tampamos dois fios pretos juntos.
O fio vermelho não é mais usado nas duas extremidades (chave) e agora é limitado.