1 files changed, 1 insertions, 1 deletions
diff --git a/test.py b/test.py
index 53a1d4b..ad42b2d 100644
--- a/test.py
+++ b/test.py
@@ -32,7 +32,7 @@ if __name__ == '__main__':
         print('%04d: process image... %s' % (i, img_path))
         ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
         print(dataset.name())
-        if dataset.name() == 'RecursiveImageDataset':
+        if dataset.name() == 'RecursiveDatasetDataLoader':
             # dataset.append(ims)
             print(ims)