diff --git a/frontend/src/components/notebookViewer/BlockOutput.vue b/frontend/src/components/notebookViewer/BlockOutput.vue
index e6ac565..ec6a68e 100644
--- a/frontend/src/components/notebookViewer/BlockOutput.vue
+++ b/frontend/src/components/notebookViewer/BlockOutput.vue
@@ -21,7 +21,7 @@
               <v-col cols="12" v-if="graphic" class="cell-content output-display">
                 <img v-if="'image/png' in output.data" alt="Image Error"
                      :src="b64ToUrl(normaliseJupyterOutput(output.data['image/png']),'image/png')"/>
-                <HTMLOutput v-if="'text/html' in output.data" :html="output.data['text/html'].join('')"></HTMLOutput>
+                <HTMLOutput v-if="'text/html' in output.data" :html="output.data['text/html'].join('')" />
               </v-col>
             </template>
             <pre v-else-if="output.output_type==='error'" class="cell-content output-err"
@@ -42,7 +42,7 @@
 import {Component, Prop, Vue} from "vue-property-decorator";
 import {Cell} from "@/types/shims/shims-nbformat-v4";
 import Convert from "ansi-to-html";
-import HTMLOutput from "./HTMLOutput.vue";
+import HTMLOutput from "@/components/notebookViewer/HTMLOutput.vue";
 
 const convert = new Convert();
 @Component
diff --git a/frontend/src/components/notebookViewer/HTMLOutput.vue b/frontend/src/components/notebookViewer/HTMLOutput.vue
index 3523081..76b756f 100644
--- a/frontend/src/components/notebookViewer/HTMLOutput.vue
+++ b/frontend/src/components/notebookViewer/HTMLOutput.vue
@@ -9,7 +9,6 @@ import {Component, Prop, Vue} from "vue-property-decorator";
 export default class HTMLOutput extends Vue {
   name = "HTMLOutput"
   @Prop(String) readonly html!: string;
-  htmlDiv!: HTMLDivElement;
 
   recurseDescendants (node: Element) {
     for (var i = 0; i < node.children.length; i++) {
@@ -24,11 +23,10 @@ export default class HTMLOutput extends Vue {
     }
   }
 
-
   mounted() {
     console.log("running scripts");
     this.$nextTick(() => {
-        this.recurseDescendants(this.htmlDiv);
+        this.recurseDescendants(this.$refs.htmlDiv as Element);
     });
   }
 }