power driver fix

1 files changed, 7 insertions, 1 deletions

diff --git a/fw/fe310/eos/net/pwr.c b/fw/fe310/eos/net/pwr.c
index 89f1817..308a05a 100644
--- a/fw/fe310/eos/net/pwr.c
+++ b/fw/fe310/eos/net/pwr.c
@@ -8,8 +8,10 @@
 #include "soc/pwr.h"
 #include "soc/spi.h"
 #include "dev/spi.h"
+#include "dev/net.h"
 #include "dev/lcd.h"
 #include "eve/eve.h"
+#include "dev/flash.h"
 
 #include "pwr.h"
 
@@ -51,7 +53,11 @@ static void pwr_handle_btn(unsigned char type, unsigned char *buffer, uint16_t l
         eos_spi_deselect();
     }
 
-    rv = eos_pwr_sleep();
+    rv = eos_net_sleep(1000);
+
+    eos_flash_norm();
+
+    eos_pwr_sleep();
 }
 
 void eos_pwr_net_init(void) {