int kvm_arch_hardware_setup(void *opaque)
 {
-       gmap_notifier.notifier_call = kvm_gmap_notifier;
-       gmap_register_pte_notifier(&gmap_notifier);
-       vsie_gmap_notifier.notifier_call = kvm_s390_vsie_gmap_notifier;
-       gmap_register_pte_notifier(&vsie_gmap_notifier);
-       atomic_notifier_chain_register(&s390_epoch_delta_notifier,
-                                      &kvm_clock_notifier);
        return 0;
 }
 
 void kvm_arch_hardware_unsetup(void)
 {
-       gmap_unregister_pte_notifier(&gmap_notifier);
-       gmap_unregister_pte_notifier(&vsie_gmap_notifier);
-       atomic_notifier_chain_unregister(&s390_epoch_delta_notifier,
-                                        &kvm_clock_notifier);
+
 }
 
 static void allow_cpu_feat(unsigned long nr)
        if (rc)
                goto err_gib;
 
+       gmap_notifier.notifier_call = kvm_gmap_notifier;
+       gmap_register_pte_notifier(&gmap_notifier);
+       vsie_gmap_notifier.notifier_call = kvm_s390_vsie_gmap_notifier;
+       gmap_register_pte_notifier(&vsie_gmap_notifier);
+       atomic_notifier_chain_register(&s390_epoch_delta_notifier,
+                                      &kvm_clock_notifier);
+
        return 0;
 
 err_gib:
 
 void kvm_arch_exit(void)
 {
+       gmap_unregister_pte_notifier(&gmap_notifier);
+       gmap_unregister_pte_notifier(&vsie_gmap_notifier);
+       atomic_notifier_chain_unregister(&s390_epoch_delta_notifier,
+                                        &kvm_clock_notifier);
+
        kvm_s390_gib_destroy();
        if (IS_ENABLED(CONFIG_VFIO_PCI_ZDEV_KVM))
                kvm_s390_pci_exit();