diff --git a/flashutils/flashutils.c b/flashutils/flashutils.c
index 2f8da41..7fa38a9 100644
--- a/flashutils/flashutils.c
+++ b/flashutils/flashutils.c
@@ -70,6 +70,17 @@
     return (pid == -1 ? -1 : pstat);
 }
 
+int get_flash_type(const char* partitionType) {
+    int type = UNSUPPORTED;
+    if (strcmp(partitionType, "mtd") == 0)
+        type = MTD;
+    else if (strcmp(partitionType, "emmc") == 0)
+        type = MMC;
+    else if (strcmp(partitionType, "bml") == 0)
+        type = BML;
+    return type;
+}
+
 static int detect_partition(const char *partitionType, const char *partition)
 {
     int type = device_flash_type();
@@ -81,12 +92,7 @@
         type = BML;
 
     if (partitionType != NULL) {
-        if (strstr(partitionType, "mtd") != NULL)
-            type = MTD;
-        else if (strstr(partitionType, "emmc") != NULL)
-            type = MMC;
-        else if (strstr(partitionType, "bml") != NULL)
-            type = BML;
+        type = get_flash_type(partitionType);
     }
 
     printf("partitionType: %s\n", partitionType);