2 files changed, 5 insertions, 4 deletions
diff --git a/src/kernel/fs/ata_pio.c b/src/kernel/fs/ata_pio.c
index b6dcdcc..ef9a23e 100644
--- a/src/kernel/fs/ata_pio.c
+++ b/src/kernel/fs/ata_pio.c
@@ -81,7 +81,7 @@ uint8_t *ata_read28(struct ata_interface *interface, uint32_t sector)
 uint8_t ata_write28(struct ata_interface *interface, uint32_t sector, const uint8_t *contents)
 {
     if (sector > 0x0FFFFFFF) return 1;
-    asm ("cli");
+    cli();
 
     outb(interface->device_port, (uint8_t) ((interface->master ? 0xE0 : 0xF0) | ((sector & 0x0F000000) >> 24)));
 
@@ -122,4 +122,4 @@ uint8_t ata_clear28(struct ata_interface *interface, uint32_t sector)
 {
     uint8_t empty_sector[512] = {0};
     return ata_write28(interface, sector, empty_sector);
-}
-\ No newline at end of file
+}
diff --git a/src/kernel/fs/install.c b/src/kernel/fs/install.c
index 84d74fd..0c3d824 100644
--- a/src/kernel/fs/install.c
+++ b/src/kernel/fs/install.c
@@ -1,5 +1,6 @@
 #include <kernel/graphics/vesa.h>
 #include <kernel/fs/ata_pio.h>
+#include <kernel/io/io.h>
 #include <kernel/fs/marfs/marfs.h>
 #include <kernel/fs/iso9660/iso9660.h>
 #include <kernel/fs/atapi_pio.h>
@@ -12,7 +13,7 @@
 void install_melvix()
 {
     info("You're booting from a CD, Melvix will only run after an installation");
-    asm ("cli");
+    cli();
     struct ata_interface *primary_master = new_ata(1, 0x1F0);
     if (marfs_init(primary_master) != 0) {
         panic("No HDD found!");
@@ -90,4 +91,4 @@ void install_melvix()
     // timer_wait(200);
     acpi_poweroff();
     halt_loop();
-}
-\ No newline at end of file
+}