diff --git a/rust/src/main.rs b/rust/src/main.rs
index 3f3b1ad..4bb183f 100644
--- a/rust/src/main.rs
+++ b/rust/src/main.rs
@@ -1081,15 +1081,15 @@ fn get_version() -> VersionInfo {
 
     let running_partition = unsafe { esp_ota_get_running_partition() };
     let address = unsafe { (*running_partition).address };
-    let partition = if address > 20000 {
-        "ota_1"
+    let partition = if address > 100000 {
+        "ota_1 @ "
     } else {
-        "ota_0"
+        "ota_0 @ "
     };
     return VersionInfo {
         git_hash: (branch + "@" + hash),
         build_time: env!("VERGEN_BUILD_TIMESTAMP").to_owned(),
-        partition: partition.to_owned()
+        partition: partition.to_owned() + &address.to_string()
     };
 }