diff --git a/common/recipes-phosphor/flash/openpower-software-manager.bb b/common/recipes-phosphor/flash/openpower-software-manager.bb
index fa4a20f..28949a8 100644
--- a/common/recipes-phosphor/flash/openpower-software-manager.bb
+++ b/common/recipes-phosphor/flash/openpower-software-manager.bb
@@ -28,16 +28,13 @@
 
 SRC_URI += "git://github.com/openbmc/openpower-pnor-code-mgmt"
 
-SRC_URI += " \
-        file://bios-ubiattach \
-        file://bios-ubiformat"
+SRC_URI += "file://obmc-flash-bios"
 
 SRCREV = "a8eda56300d309cb11566787d70fd1b9a5721e07"
 
 do_install_append() {
         install -d ${D}${sbindir}
-        install -m 0755 ${WORKDIR}/bios-ubiattach ${D}${sbindir}/bios-ubiattach
-        install -m 0755 ${WORKDIR}/bios-ubiformat ${D}${sbindir}/bios-ubiformat
+        install -m 0755 ${WORKDIR}/obmc-flash-bios ${D}${sbindir}/obmc-flash-bios
 }
 
 DBUS_SERVICE_${PN} += "org.open_power.Software.Host.Updater.service"
diff --git a/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiattach b/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiattach
deleted file mode 100644
index accf7d8..0000000
--- a/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiattach
+++ /dev/null
@@ -1,23 +0,0 @@
-#!/bin/sh
-
-# Get the pnor mtd device number
-pnor=`grep pnor /proc/mtd |cut -c 4`
-
-# Attach the pnor mtd device to ubi
-ubiattach /dev/ubi_ctrl -m $pnor
-if [ $? -ne 0 ]
-then
-  # Check the pnor mtd device is formatted as ubi by reading the first 3 byes,
-  # which should be the ascii chars 'UBI'
-  magic=`hexdump -C -n 3 /dev/mtd$pnor`
-  if [[ $magic == *"UBI"* ]]
-  then
-    # Device already formatted as ubi, ubiattach failed for some other reason
-    exit -1
-  else
-    # Format device as ubi
-    bios-ubiformat
-    # Retry the ubiattach
-    ubiattach /dev/ubi_ctrl -m $pnor
-  fi
-fi
diff --git a/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiformat b/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiformat
deleted file mode 100644
index d62a6b2..0000000
--- a/common/recipes-phosphor/flash/openpower-software-manager/bios-ubiformat
+++ /dev/null
@@ -1,8 +0,0 @@
-#!/bin/sh
-
-# Get the mtd device number for pnor
-pnor=`grep pnor /proc/mtd |cut -c 4`
-
-# Format the pnor mtd device as ubi
-echo "Starting ubiformat /dev/mtd$pnor"
-ubiformat /dev/mtd$pnor -y -q
diff --git a/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios b/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios
new file mode 100644
index 0000000..acc550b
--- /dev/null
+++ b/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios
@@ -0,0 +1,49 @@
+#!/bin/sh
+
+# Get the mtd device number (mtdX)
+findmtd() {
+  m="$(grep -xl "$1" /sys/class/mtd/*/name)"
+  m="${m%/name}"
+  m="${m##*/}"
+  echo "${m}"
+}
+
+# Attach the pnor mtd device to ubi
+attach_ubi() {
+  pnormtd="$(findmtd pnor)"
+  pnor="${pnormtd#mtd}"
+  pnordev="/dev/mtd${pnor}"
+
+  ubiattach /dev/ubi_ctrl -m "${pnor}" -d "${pnor}"
+  rc=$?
+  if [ ${rc} -ne 0 ]; then
+    # Check the pnor mtd device is formatted as ubi by reading the first 3 byes,
+    # which should be the ascii chars 'UBI'
+    magic="$(hexdump -C -n 3 ${pnordev})"
+    if [[ "${magic}" =~ "UBI" ]]; then
+      # Device already formatted as ubi, ubiattach failed for some other reason
+      return ${rc}
+    else
+      # Format device as ubi
+      echo "Starting ubiformat ${pnordev}"
+      ubiformat "${pnordev}" -y -q
+      # Retry the ubiattach
+      ubiattach /dev/ubi_ctrl -m "${pnor}" -d "${pnor}"
+    fi
+  fi
+}
+
+case "$1" in
+  ubiattach)
+    attach_ubi
+    ;;
+  *)
+    echo "Invalid argument"
+    exit 1
+    ;;
+esac
+rc=$?
+if [ ${rc} -ne 0 ]; then
+  echo "$0: error ${rc}"
+  exit ${rc}
+fi
diff --git a/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios-ubiattach.service b/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios-ubiattach.service
index f90c437..bae633d 100644
--- a/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios-ubiattach.service
+++ b/common/recipes-phosphor/flash/openpower-software-manager/obmc-flash-bios-ubiattach.service
@@ -4,5 +4,4 @@
 [Service]
 Type=oneshot
 RemainAfterExit=yes
-ExecStart={sbindir}/bios-ubiattach
-ExecStop=/bin/sh -c 'grep pnor /proc/mtd |cut -c 4 |xargs {sbindir}/ubidetach /dev/ubi_ctrl -m'
+ExecStart={sbindir}/obmc-flash-bios ubiattach