diff --git a/src/AgOledDisplay.cpp b/src/AgOledDisplay.cpp
index ccce42a..bd8ee95 100644
--- a/src/AgOledDisplay.cpp
+++ b/src/AgOledDisplay.cpp
@@ -357,7 +357,7 @@ void OledDisplay::showFirmwareUpdateProgress(int percent) {
   } while (DISP()->nextPage());
 }
 
-void OledDisplay::showFirmwareUpdateSuccess(String count) {
+void OledDisplay::showFirmwareUpdateSuccess(int count) {
   if (isDisplayOff) {
     return;
   }
@@ -367,7 +367,7 @@ void OledDisplay::showFirmwareUpdateSuccess(String count) {
     DISP()->setFont(u8g2_font_t0_16_tf);
     setCentralText(20, "Firmware Update");
     setCentralText(40, "Success");
-    setCentralText(60, String("Rebooting... ") + count);
+    setCentralText(60, String("Rebooting... ") + String(count));
   } while (DISP()->nextPage());
 }
 
diff --git a/src/AgOledDisplay.h b/src/AgOledDisplay.h
index 1013935..4073f0f 100644
--- a/src/AgOledDisplay.h
+++ b/src/AgOledDisplay.h
@@ -38,7 +38,7 @@ public:
   void setBrightness(int percent);
   void showFirmwareUpdateVersion(String version);
   void showFirmwareUpdateProgress(int percent);
-  void showFirmwareUpdateSuccess(String count);
+  void showFirmwareUpdateSuccess(int count);
   void showFirmwareUpdateFailed(void);
   void showFirmwareUpdateSkipped(void);
   void showFirmwareUpdateUpToDate(void);