diff --git a/src/mlpack/tests/callback_test.cpp b/src/mlpack/tests/callback_test.cpp
index e66fc1e051..0d532df67c 100644
--- a/src/mlpack/tests/callback_test.cpp
+++ b/src/mlpack/tests/callback_test.cpp
@@ -42,9 +42,9 @@ TEST_CASE("FFNCallbackTest", "[CallbackTest]")
   arma::mat data;
   arma::mat labels;
 
-  if (!data::Load("lab1.csv", data, true))
+  if (!data::Load("lab1.csv", data))
     FAIL("Cannot load test dataset lab1.csv!");
-  if (!data::Load("lab3.csv", labels, true))
+  if (!data::Load("lab3.csv", labels))
     FAIL("Cannot load test dataset lab3.csv!");
 
   FFN<MeanSquaredError<>, RandomInitialization> model;
@@ -68,9 +68,9 @@ TEST_CASE("FFNWithOptimizerCallbackTest", "[CallbackTest]")
   arma::mat data;
   arma::mat labels;
 
-  if (!data::Load("lab1.csv", data, true))
+  if (!data::Load("lab1.csv", data))
     FAIL("Cannot load test dataset lab1.csv!");
-  if (!data::Load("lab3.csv", labels, true))
+  if (!data::Load("lab3.csv", labels))
     FAIL("Cannot load test dataset lab3.csv!");
 
   FFN<MeanSquaredError<>, RandomInitialization> model;
diff --git a/src/mlpack/tests/svd_batch_test.cpp b/src/mlpack/tests/svd_batch_test.cpp
index bac6df4917..41b005a28d 100644
--- a/src/mlpack/tests/svd_batch_test.cpp
+++ b/src/mlpack/tests/svd_batch_test.cpp
@@ -70,7 +70,8 @@ class SpecificRandomInitialization
 TEST_CASE("SVDBatchMomentumTest", "[SVDBatchTest]")
 {
   mat dataset;
-  data::Load("GroupLensSmall.csv", dataset);
+  if (!data::Load("GroupLensSmall.csv", dataset))
+    FAIL("Cannot load dataset GroupLensSmall.csv!");
 
   // Generate list of locations for batch insert constructor for sparse
   // matrices.
@@ -117,7 +118,8 @@ TEST_CASE("SVDBatchMomentumTest", "[SVDBatchTest]")
 TEST_CASE("SVDBatchRegularizationTest", "[SVDBatchTest]")
 {
   mat dataset;
-  data::Load("GroupLensSmall.csv", dataset);
+  if (!data::Load("GroupLensSmall.csv", dataset))
+    FAIL("Cannot load dataset GroupLensSmall.csv!");
 
   // Generate list of locations for batch insert constructor for sparse
   // matrices.