KVM: arm64: Adjust partial code of hyp stage-1 map and guest stage-2 map