diff --git a/src/data/DataProvider.cpp b/src/data/DataProvider.cpp
index cdac0cda44a116005800cb306b078887c1ed72c4..ab94547ec40431070d30311c3d04e5227d18b350 100644
--- a/src/data/DataProvider.cpp
+++ b/src/data/DataProvider.cpp
@@ -44,7 +44,7 @@ std::vector<std::shared_ptr<Tensor>> DataProvider::readBatch(size_t startIndex)
         batchData->resize(dataBatchSize[i]);
         // batchData->setBackend(mDataBackends[i]);
         batchData->setBackend("cpu");
-        batchData->setDatatype(mDataTypes[i]);
+        batchData->setDataType(mDataTypes[i]);
         batchTensors.push_back(batchData);
     }