diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 83e3d8ea2d7b965d6327e9ca575b72cfd8b2d305..9a7b6d75fec87c22945a5e6ebec3af76241c4c31 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -5,27 +5,27 @@ image:
 variables:
   GIT_SUBMODULE_STRATEGY: recursive
 
-read_test:
-  stage: build
-  script:
-  - mkdir -p result
-  - make
-  - ./testbench_ecp5_minifpga.bin &
-  - sleep 1s && python3 src/py/tty1.py
-  artifacts:
-    paths:
-    - result
+#read_test:
+#  stage: build
+#  script:
+#  - mkdir -p result
+#  - make
+#  - ./testbench_ecp5_minifpga.bin &
+#  - sleep 1s && python3 src/py/tty1.py
+#  artifacts:
+#    paths:
+#    - result
 
-write_test:
-  stage: build
-  script:
-  - mkdir -p result
-  - make
-  - ./testbench_ecp5_minifpga.bin &
-  - sleep 1s && python3 src/py/tty2.py
-  artifacts:
-    paths:
-    - result
+#write_test:
+#  stage: build
+#  script:
+#  - mkdir -p result
+#  - make
+#  - ./testbench_ecp5_minifpga.bin &
+#  - sleep 1s && python3 src/py/tty2.py
+#  artifacts:
+#    paths:
+#    - result
 
 train_test:
   stage: build
diff --git a/firmware/firmware.c b/firmware/firmware.c
index 0d1fcacc96f449617deab4877160d2ac5672721d..bf107cedd144d0545483ebfad123b11f0c09161f 100644
--- a/firmware/firmware.c
+++ b/firmware/firmware.c
@@ -304,7 +304,7 @@ void main()
 
             case TRAIN_RUN_EPOCHS:
                 uint32_t num_epochs = atoi(numstr);
-                //response = run_training(response, num_epochs, learning_rate, decay_rate, token_series, token_counter);
+                response = run_training(response, num_epochs, learning_rate, decay_rate, token_series, token_counter);
                 command_mode = START;
                 break;