diff --git a/data/pascal-voc-classes.txt b/data/pascal-voc-classes.txt
deleted file mode 100644
index 008dd5fb..00000000
--- a/data/pascal-voc-classes.txt
+++ /dev/null
@@ -1,21 +0,0 @@
-background
-aeroplane
-bicycle
-bird
-boat
-bottle
-bus
-car
-cat
-chair
-cow
-diningtable
-dog
-horse
-motorbike
-person
-pottedplant
-sheep
-sofa
-train
-tvmonitor
\ No newline at end of file
diff --git a/dataloader/dataloader_impl.hpp b/dataloader/dataloader_impl.hpp
index 7b1c2636..d2fe3674 100644
--- a/dataloader/dataloader_impl.hpp
+++ b/dataloader/dataloader_impl.hpp
@@ -194,6 +194,12 @@ template<
   // Read the xml file.
   for (boost::filesystem::path annotationFile : annotationsDirectory)
   {
+    if (annotationFile.string().length() <= 3 ||
+        annotationFile.string().substr(
+            annotationFile.string().length() - 3) != "xml")
+    {
+      continue;
+    }
     // Read the xml file.
     boost::property_tree::ptree xmlFile;
     boost::property_tree::read_xml(annotationFile.string(), xmlFile);
diff --git a/tests/augmentation_tests.cpp b/tests/augmentation_tests.cpp
index 37325eed..4ac52366 100644
--- a/tests/augmentation_tests.cpp
+++ b/tests/augmentation_tests.cpp
@@ -19,7 +19,8 @@ BOOST_AUTO_TEST_SUITE(AugmentationTest);
 
 BOOST_AUTO_TEST_CASE(ResizeAugmentationTest)
 {
-  Augmentation<> augmentation(std::vector<std::string>(1, "resize (5, 4)"), 0.2);
+  Augmentation<> augmentation(std::vector<std::string>(1,
+      "resize (5, 4)"), 0.2);
 
   // Test on a square matrix.
   arma::mat input;
diff --git a/tests/dataloader_tests.cpp b/tests/dataloader_tests.cpp
index 53b24b7a..ec3ef100 100644
--- a/tests/dataloader_tests.cpp
+++ b/tests/dataloader_tests.cpp
@@ -94,9 +94,33 @@ BOOST_AUTO_TEST_CASE(MNISTDataLoaderTest)
 BOOST_AUTO_TEST_CASE(ObjectDetectionDataLoader)
 {
   DataLoader<> dataloader;
-  dataloader.LoadObjectDetectionDataset("./../data/annotations/",
-      "./../data/images/", {"person", "foot", "aeroplane", "head", "hand"},
-      {"resize 64, 64"});
+  Utils::ExtractFiles("./../data/PASCAL-VOC-Test.zip", "./../data/");
+
+  // Set paths for dataset.
+  std::string basePath = "./../data/PASCAL-VOC-Test/";
+  std::string annotaionPath = "Annotations/";
+  std::string imagesPath = "Images/";
+
+  // Classes in the dataset.
+  std::vector<std::string> classes = {"background", "aeroplane", "bicycle",
+      "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow",
+      "diningtable", "dog", "horse", "motorbike", "person", "pottedplant",
+      "sheep", "sofa", "train", "tvmonitor"};
+
+  // Resize the image to 64 x 64.
+  std::vector<std::string> augmentation = {"resize (64, 64)"};
+  dataloader.LoadObjectDetectionDataset(basePath + annotaionPath,
+      basePath + imagesPath, classes, augmentation);
+
+  // There are total 15 objects in images.
+  BOOST_REQUIRE_EQUAL(dataloader.TrainLabels().n_cols, 15);
+  // They correspond to class name, x1, y1, x2, y2.
+  BOOST_REQUIRE_EQUAL(dataloader.TrainLabels().n_rows, 5);
+
+  // Rows will be equal to shape image depth * image width * image height.
+  BOOST_REQUIRE_EQUAL(dataloader.TrainFeatures().n_rows, 64 * 64 * 3);
+  // There are total 15 objects in the images.
+  BOOST_REQUIRE_EQUAL(dataloader.TrainFeatures().n_cols, 15);
 }
 
 BOOST_AUTO_TEST_SUITE_END();