diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
index 5fce5783..c9838bf6 100644
--- a/.github/workflows/docs.yaml
+++ b/.github/workflows/docs.yaml
@@ -2,6 +2,9 @@ name: Manually publish docs
 
 on:
   workflow_dispatch:
+  push:
+    branches:
+      - main
 
 jobs:
   docs:
diff --git a/src/docs/docs-components/ThemeBuilder/Customiser/Value.jsx b/src/docs/docs-components/ThemeBuilder/Customiser/Value.jsx
index 1b2973dc..96221c68 100644
--- a/src/docs/docs-components/ThemeBuilder/Customiser/Value.jsx
+++ b/src/docs/docs-components/ThemeBuilder/Customiser/Value.jsx
@@ -1,5 +1,6 @@
 import React, { useState } from 'react';
-import { cloneDeep, set } from 'es-toolkit';
+import { cloneDeep } from 'es-toolkit';
+import { set } from 'es-toolkit/compat';
 
 import ColourPicker from './ColourPicker.jsx';