diff --git a/packages/jspsych/src/modules/plugin-api/SimulationAPI.ts b/packages/jspsych/src/modules/plugin-api/SimulationAPI.ts
index 94288c16..43f5990f 100644
--- a/packages/jspsych/src/modules/plugin-api/SimulationAPI.ts
+++ b/packages/jspsych/src/modules/plugin-api/SimulationAPI.ts
@@ -1,6 +1,8 @@
 export class SimulationAPI {
+  constructor(private getDisplayContainerElement: () => HTMLElement) {}
+
   dispatchEvent(event: Event) {
-    document.body.dispatchEvent(event);
+    this.getDisplayContainerElement().dispatchEvent(event);
   }
 
   /**