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 61cfcaaba..216d3ea97 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
@@ -452,7 +452,7 @@ object OneDAL {
       val numCols = list(0).getAs[Vector](1).toArray.size
 
       val labelsArray = new Array[Double](numRows)
-      val featuresAddress= OneDAL.cNewFloatArray(numRows.toLong * numCols)
+      val featuresAddress= OneDAL.cNewDoubleArray(numRows.toLong * numCols)
       for ( i <- 0 until  numberCores) {
         val f = Future {
           val iter = list.iterator
@@ -626,7 +626,7 @@ object OneDAL {
       val numRows = list.size
       val numCols = list(0).toArray.size
       val size = numRows.toLong * numCols.toLong
-      val targetArrayAddress = OneDAL.cNewFloatArray(size)
+      val targetArrayAddress = OneDAL.cNewDoubleArray(size)
       for ( i <- 0 until  numberCores) {
         val f = Future {
           val iter = list.iterator