diff --git a/src/stores/simulation.js b/src/stores/simulation.js
index 8595573..7f92e03 100644
--- a/src/stores/simulation.js
+++ b/src/stores/simulation.js
@@ -76,7 +76,7 @@ export const useSimulationStore = defineStore("simulation", () => {
 
     const loadWorkspace = (index) => {
         Blockly.Events.disable();
-        const workspace = mainWorkspace.value;
+        const workspace = mainWorkspace.value ?? Blockly.getMainWorkspace();
         const data = particle_array.value[index]?.blockly_workspace;
         if (data) {
             Blockly.serialization.workspaces.load(data, workspace);
@@ -100,14 +100,14 @@ export const useSimulationStore = defineStore("simulation", () => {
     }
 
     const regenerateCode = () => {
-        const ws = mainWorkspace.value;
+        const ws = mainWorkspace.value ?? Blockly.getMainWorkspace();
         const particle_base = ws.getBlocksByType("particle_base")[0];
         generated_code.value = jsonGenerator.blockToCode(particle_base);
         particle_array.value[selected_particle.value].update_data(JSON.parse(generated_code.value));
     }
 
     const saveWorkspace = (index) => {
-        const json = Blockly.serialization.workspaces.save(mainWorkspace.value);
+        const json = Blockly.serialization.workspaces.save(mainWorkspace.value ?? Blockly.getMainWorkspace());
         particle_array.value[index].blockly_workspace = json;
     }
 
@@ -141,7 +141,6 @@ export const useSimulationStore = defineStore("simulation", () => {
         debug,
         canvas_size,
         particle_array_length,
-        mainWorkspace,
         addParticle,
         removeParticle,
         removeSelectedParticle,