A direção ficar pesada depois de desligar a tração é normal e de uma forma ou outra isso acontece em todos veículos 4x4 e acontece porque mesmo com sinalização no painel indicando 4x2 a tração 4x4 fica engatada até que acabe a tensão na transmissão.
Acontece a mesma coisa com veículos como a Cherokee que mesmo com a alavanca já em 4x2 continua pesada com a tração ligada até que a tensão acabe. Em outros é impossível mover a alavanca de 4x4 para 4x2 de a tração estiver sob tensão.
Antes que alguém diga que o sistema elétrico é muito pior, é bom lembrar primeiro que as picapes L200 e Hilux que tem acionamento por alavanca tem o acionamento da roda livre elétrica.