diff --git a/src/bootloader/bootloader_with_flash.c b/src/bootloader/bootloader_with_flash.c
index 990221c..4f78a61 100644
--- a/src/bootloader/bootloader_with_flash.c
+++ b/src/bootloader/bootloader_with_flash.c
@@ -230,12 +230,14 @@ int main(int argc, char *argv[])
 	uint32_t img_size, block_sel;
 
 	volatile uint32_t *pGPIO_DATA = (uint32_t*)SYS_GPIO_0_DATA;
+	volatile uint32_t *pGPIO_DIR = (uint32_t*)SYS_GPIO_0_DIR;
 	volatile uint8_t *pMem;
 	volatile uint32_t *pFlashIO = (uint32_t*)FLASH_BASE_IO;		// Flash in uncached area
 	volatile uint32_t *pFlashMem = (uint32_t*)FLASH_BASE_MEM;	// Flash in cached area
 	volatile uint32_t *ram32 = (uint32_t*)SDRAM_BASE;
 	uint32_t flash_offset;
 
+	*pGPIO_DIR  = GPIO_0_DFLT_DIR;
 	*pGPIO_DATA = GPIO_0_DFLT_DATA;
 	haddr = 0;
 	block_sel = 0x7F & (uint32_t)((*pGPIO_DATA & GPIO_0_DIP) >> GPIO_0_ALIGN_DIP);