diff --git a/src/store.js b/src/store.js
index 4804dcd2..f622f88b 100644
--- a/src/store.js
+++ b/src/store.js
@@ -60,6 +60,7 @@ export function saveConfig(key, value) {
 // render the code if there are fetched files for current selected template
 export function genCode() {
   const currentFiles = files[store.config.template]
+  store.code = {}
   if (currentFiles && Object.keys(currentFiles).length) {
     for (const file in currentFiles) {
       if (!store.config.include_test && file === 'test_all.py') {
diff --git a/src/templates/templates.json b/src/templates/templates.json
index fd7b81ed..41f6efa8 100644
--- a/src/templates/templates.json
+++ b/src/templates/templates.json
@@ -35,8 +35,8 @@
   "template-text-classification": [
     "README.md",
     "config.yaml",
-    "main.py",
     "data.py",
+    "main.py",
     "model.py",
     "trainers.py",
     "utils.py",