diff --git a/src/logic/sourceMapping.tsx b/src/logic/sourceMapping.tsx
index 0db53d7..f4b5240 100644
--- a/src/logic/sourceMapping.tsx
+++ b/src/logic/sourceMapping.tsx
@@ -1,6 +1,7 @@
 import { Token } from "../types/prism-types";
 
 export interface TMappingELement {
+  token: Token;
   start: number;
   end: number;
   ref: { current: HTMLElement | null };
@@ -20,6 +21,7 @@ export const sourceMapping: TSourceMapping = {
   lastEnd: 0,
   registerToken(t: Token) {
     const obj: TMappingELement = {
+      token: t,
       start: this.lastEnd,
       end: this.lastEnd + t.content.length,
       ref: { current: null },
@@ -27,8 +29,6 @@ export const sourceMapping: TSourceMapping = {
     this.mappingArray.push(obj);
     this.lastEnd = this.mappingArray[this.mappingArray.length - 1].end;
 
-    console.log(obj, t);
-
     return function updateHTMLElement(ele) {
       obj.ref.current = ele;
     };