hr->dawrx1 = swab64(hr->dawrx1);
 }
 
-static void save_hv_return_state(struct kvm_vcpu *vcpu, int trap,
+static void save_hv_return_state(struct kvm_vcpu *vcpu,
                                 struct hv_guest_state *hr)
 {
        struct kvmppc_vcore *vc = vcpu->arch.vcore;
        hr->pidr = vcpu->arch.pid;
        hr->cfar = vcpu->arch.cfar;
        hr->ppr = vcpu->arch.ppr;
-       switch (trap) {
+       switch (vcpu->arch.trap) {
        case BOOK3S_INTERRUPT_H_DATA_STORAGE:
                hr->hdar = vcpu->arch.fault_dar;
                hr->hdsisr = vcpu->arch.fault_dsisr;
        delta_spurr = vcpu->arch.spurr - l2_hv.spurr;
        delta_ic = vcpu->arch.ic - l2_hv.ic;
        delta_vtb = vc->vtb - l2_hv.vtb;
-       save_hv_return_state(vcpu, vcpu->arch.trap, &l2_hv);
+       save_hv_return_state(vcpu, &l2_hv);
 
        /* restore L1 state */
        vcpu->arch.nested = NULL;