diff --git a/src/store/sourceFiles/reducer.ts b/src/store/sourceFiles/reducer.ts
index c7a0589..2e80ef7 100644
--- a/src/store/sourceFiles/reducer.ts
+++ b/src/store/sourceFiles/reducer.ts
@@ -18,7 +18,8 @@ export function sourceFilesReducer(
     case INIT_SOURCEFILE:
       return {
         ...state,
-        inititalSourceFile: action.payload.sourceFile
+        inititalSourceFile: action.payload.sourceFile,
+        currentSourceFile: action.payload.sourceFile
       };
     case UPDATE_SOURCEFILE:
       return {