From 003492cd7c9c7d35e715907bab33085bf4d19093 Mon Sep 17 00:00:00 2001
From: Valeriy S <ValeraS@users.noreply.github.com>
Date: Tue, 4 Dec 2018 08:34:48 +0300
Subject: [PATCH] fix(client): restore only the contents of the challenge files

---
 .../Challenges/redux/code-storage-epic.js         | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/client/src/templates/Challenges/redux/code-storage-epic.js b/client/src/templates/Challenges/redux/code-storage-epic.js
index cb0470b296..8cc3b6b76e 100644
--- a/client/src/templates/Challenges/redux/code-storage-epic.js
+++ b/client/src/templates/Challenges/redux/code-storage-epic.js
@@ -98,7 +98,20 @@ function loadCodeEpic(action$, state$) {
 
       const codeFound = getCode(id);
       if (codeFound && isFilesAllPoly(codeFound)) {
-        finalFiles = codeFound;
+        finalFiles = {
+          ...fileKeys.map(key => files[key]).reduce(
+            (files, file) => ({
+              ...files,
+              [file.key]: {
+                ...file,
+                contents: codeFound[file.key]
+                  ? codeFound[file.key].contents
+                  : file.contents
+              }
+            }),
+            {}
+          )
+        };
       } else {
         const legacyCode = getLegacyCode(legacyKey);
         if (legacyCode && !invalidForLegacy) {