1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/int.S b/src/core/int.S
index 2897698..c8612dc 100644
--- a/src/core/int.S
+++ b/src/core/int.S
@@ -26,6 +26,9 @@
     push %rbp
     push %rsi
     push %rdi
+    mov %ds, %rax
+    push %rax
+    mov 56(%rsp), %rax # Restore %rax
     push %rax
     mov $0x10, %ax
     mov %ax, %ds
@@ -33,6 +36,8 @@
 .endm
 
 .macro RESTORE_REGS
+    pop %rax
+    mov %rax,%ds
     pop %rdi    
     pop %rsi    
     pop %rbp