1 files changed, 2 insertions, 3 deletions
diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c
index ac5fc6c64b7e..bfe4a57bcc10 100755..100644
--- a/virt/kvm/kvm_main.c
+++ b/virt/kvm/kvm_main.c
@@ -67,13 +67,12 @@
 
 #include <linux/kvm_dirty_ring.h>
 
-#include <linux/sev-step.h>
-#include <linux/userspace_page_track_signals.h>
-
 /* Worst case buffer size needed for holding an integer. */
 #define ITOA_MAX_LEN 12
 
 #include "../../arch/x86/kvm/svm/cachepc/kvm.h"
+#include "../../arch/x86/kvm/sevstep/sevstep.h"
+#include "../../arch/x86/kvm/sevstep/uspt.h"
 
 MODULE_AUTHOR("Qumranet");
 MODULE_LICENSE("GPL");