diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 2427a817341b7339f5515d9a616d36932145813c..cbdb6fc7414c3316f7a1eac9ecba82f39efbcb31 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,26 +1,20 @@
 stages:
+  - cpplint
   - ops_test
   - ops_benchmark
-  - cpplint
 
 cpplint:
   stage: cpplint
-  only:
-    - master
   script:
     - curl -o cpplint.py https://raw.githubusercontent.com/google/styleguide/gh-pages/cpplint/cpplint.py
     - python cpplint.py --linelength=80 --counting=detailed $(find mace -name *.h -or -name *.cc)
 
 ops_test:
   stage: ops_test
-  only:
-    - master
   script:
     - FAILURE_PATTERN="FAILED\|Aborted" tools/bazel-adb-run.sh //mace/ops:ops_test
 
 ops_benchmark:
   stage: ops_benchmark
-  only:
-    - master
   script:
     - FAILURE_PATTERN="Aborted" tools/bazel-adb-run.sh //mace/ops:ops_benchmark --pattern=.*CONV.*
diff --git a/mace/core/testing/test_benchmark_main.cc b/mace/core/testing/test_benchmark_main.cc
index 9c2d1daa916513054796ec8ad22caca59869c772..079a79e79afb35ed3319380409ae6155ec4112d9 100644
--- a/mace/core/testing/test_benchmark_main.cc
+++ b/mace/core/testing/test_benchmark_main.cc
@@ -7,6 +7,7 @@
 #include "gflags/gflags.h"
 #include "mace/core/testing/test_benchmark.h"
 #include "mace/public/mace.h"
+#include "mace/public/mace_runtime.h"
 
 DEFINE_string(pattern, "all", "op benchmark pattern, eg:.*CONV.*");
 DEFINE_int32(gpu_perf_hint, 3, "0:DEFAULT/1:LOW/2:NORMAL/3:HIGH");