diff --git a/packages/plugin-bust/src/index.js b/packages/plugin-bust/src/index.js
index dc6ac188b2d..58decb66fa3 100644
--- a/packages/plugin-bust/src/index.js
+++ b/packages/plugin-bust/src/index.js
@@ -4,12 +4,8 @@ export default {
   name: name,
   version: version,
   hooks: {
-    preRender: function (svg) {
-      if (svg.attributes.get('freesewing:plugin-bust') === false) {
-        svg.attributes.set('freesewing:plugin-bust', version)
-      }
-    },
-    preDraft: function ({ settings }) {
+    preRender: (svg) => svg.attributes.setIfUnset('freesewing:plugin-bust', version),
+    preDraft: ({ settings }) => {
       if (typeof settings.measurements.bust === 'undefined') {
         settings.measurements.bust = settings.measurements.chest
         settings.measurements.chest = settings.measurements.highBust