diff --git a/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp b/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp
index d27aa4e46..ba090703e 100644
--- a/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp
+++ b/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp
@@ -656,7 +656,7 @@ TEST_F(DeclarableOpsTests9, concat_test19) {
         context.setInputArray(e, array, true);
     }
 
-    auto z = NDArrayFactory::create<float>('c', {arrays.size(), 5, 20});
+    auto z = NDArrayFactory::create<float>('c', {(Nd4jLong) arrays.size(), 5, 20});
     context.setOutputArray(0, &z, false);
     context.setIArguments(&axis, 1);
 
diff --git a/libnd4j/tests_cpu/layers_tests/testlayers.h b/libnd4j/tests_cpu/layers_tests/testlayers.h
index 723b241bf..95814faae 100644
--- a/libnd4j/tests_cpu/layers_tests/testlayers.h
+++ b/libnd4j/tests_cpu/layers_tests/testlayers.h
@@ -36,5 +36,6 @@
 #include <GraphExecutioner.h>
 #include <gtest/gtest.h>
 #include <NDArrayFactory.h>
+#include <array>
 
 #endif //LIBND4J_TESTLAYERS_H