2 files changed, 4 insertions, 7 deletions

diff --git a/firmware/target/arm/imx31/ata-imx31.c b/firmware/target/arm/imx31/ata-imx31.c
index 8a40c924ff..80539b9c9d 100644
--- a/firmware/target/arm/imx31/ata-imx31.c
+++ b/firmware/target/arm/imx31/ata-imx31.c
@@ -25,9 +25,8 @@
 #include "system.h"
 #include "power.h"
 #include "panic.h"
-#include "ata.h"
+#include "ata-driver.h"
 #include "ata-defines.h"
-#include "ata-target.h"
 #include "ccm-imx31.h"
 #ifdef HAVE_ATA_DMA
 #include "sdma-imx31.h"
diff --git a/firmware/target/arm/imx31/ata-target.h b/firmware/target/arm/imx31/ata-target.h
index 24141c6aa7..6893e83198 100644
--- a/firmware/target/arm/imx31/ata-target.h
+++ b/firmware/target/arm/imx31/ata-target.h
@@ -21,6 +21,8 @@
 #ifndef ATA_TARGET_H
 #define ATA_TARGET_H
 
+#include "config.h"
+
 #ifdef BOOTLOADER
 #define ATA_DRIVER_CLOSE
 #endif
@@ -46,12 +48,8 @@
 #define ATA_COMMAND     ATA_DRIVE_COMMAND
 #define ATA_CONTROL     ATA_DRIVE_CONTROL
 
-void ata_reset(void);
-void ata_device_init(void);
-bool ata_is_coldstart(void);
 
-#define ATA_SET_DEVICE_FEATURES
-void ata_set_pio_timings(int mode);
+#define ATA_SET_PIO_TIMING
 
 #define ATA_TARGET_POLLING