diff --git a/task-launcher/src/tasks/mental-rotation/trials/instructions.ts b/task-launcher/src/tasks/mental-rotation/trials/instructions.ts
index b1fd4377..d5f81994 100644
--- a/task-launcher/src/tasks/mental-rotation/trials/instructions.ts
+++ b/task-launcher/src/tasks/mental-rotation/trials/instructions.ts
@@ -1,7 +1,9 @@
import jsPsychAudioMultiResponse from '@jspsych-contrib/plugin-audio-multi-response';
import { mediaAssets } from '../../..';
// @ts-ignore
-import { taskStore } from '../../shared/helpers';
+import { taskStore, replayButtonSvg, PageStateHandler, setupReplayAudio } from '../../shared/helpers';
+
+const replayButtonHtmlId = 'replay-btn-revisited';
// Switch to HTMLMultiResponse when we have video with audio
export const videoInstructionsFit = {
@@ -16,6 +18,9 @@ export const videoInstructionsFit = {
prompt: () => {
return `
+