diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala
index 35fc1a00b..f9d1b4ec9 100644
--- a/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala
+++ b/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala
@@ -760,6 +760,6 @@ object OneDAL {
   @native def cNewFloatArray(size: Long): Long
 
   @native def cCopyArrayToNative(arrayAddr: Long,
-                                 data: Array[Double],
+                                 data: Array[Float],
                                  index: Long): Unit
 }