diff --git a/k/atapi.h b/k/atapi.h
index dc4ec58..64b7aef 100644
--- a/k/atapi.h
+++ b/k/atapi.h
@@ -2,13 +2,15 @@
 #define ATAPI_H
 
 #include <stdint.h>
+#include <stddef.h>
 
 struct atapi_drive
 {
-    uint8_t reg;
-    uint8_t drive;
+    uint16_t reg;
+    uint16_t drive;
 };
 
 void discover_atapi_drive(void);
+void *read_block(size_t lba);
 
 #endif /* !ATAPI_H */