diff --git a/src/operators/math/softmax.cpp b/src/operators/math/softmax.cpp
index 5ae7d96fffd3e0b4be32017630a503881c477635..eb442e634c482ce300bb87ef6d9070c9d3ff415d 100644
--- a/src/operators/math/softmax.cpp
+++ b/src/operators/math/softmax.cpp
@@ -15,7 +15,7 @@ limitations under the License. */
 #include "operators/math/softmax.h"
 #include "common/types.h"
 #if __ARM_NEON
-#include <algorithm>
+#include <math.h>
 #include "operators/math/math_func_neon.h"
 #endif