diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index 5ddf58e..ff27912 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -30,25 +30,22 @@
         },
         {
             "label": "build-all",
-            "command": "${workspaceFolder}/make.sh",
+            "command": "make",
             "args": [
-                "build",
                 "all"
             ]
         },
         {
             "label": "build-mnist",
-            "command": "${workspaceFolder}/make.sh",
+            "command": "make",
             "args": [
-                "build",
                 "mnist"
             ]
         },
         {
             "label": "build-cnn",
-            "command": "${workspaceFolder}/make.sh",
+            "command": "make",
             "args": [
-                "build",
                 "cnn"
             ]
         },
diff --git a/test/mnist_neuron_io.c b/test/mnist_neuron_io.c
index e93b95a..b36702d 100644
--- a/test/mnist_neuron_io.c
+++ b/test/mnist_neuron_io.c
@@ -1,5 +1,6 @@
 #include <stdlib.h>
 #include <stdio.h>
+#include <math.h>
 #include <stdint.h>
 #include <inttypes.h>