diff --git a/src/pstorage_platform.h b/src/pstorage_platform.h
index 75af90e..ed56e92 100644
--- a/src/pstorage_platform.h
+++ b/src/pstorage_platform.h
@@ -21,6 +21,7 @@
 
 #include <stdint.h>
 #include "nrf.h"
+#include "flash.h"
 
 static __INLINE uint16_t pstorage_flash_page_size()
 {
diff --git a/src/usb/uf2/ghostfat.c b/src/usb/uf2/ghostfat.c
index 8e8ea58..7c82a14 100644
--- a/src/usb/uf2/ghostfat.c
+++ b/src/usb/uf2/ghostfat.c
@@ -235,10 +235,6 @@ static void uf2_write_complete(uint32_t numBlocks)
   update_status.app_crc     = 0; // skip CRC checking with uf2 upgrade
   update_status.app_size    = numBlocks*256;
 
-  // re-enable softdevice
-  extern uint32_t softdev_init(bool init_softdevice);
-  softdev_init(false);
-
   bootloader_dfu_update_process(update_status);
 }