diff --git a/datasets.py b/datasets.py
index 443c786..33148d6 100644
--- a/datasets.py
+++ b/datasets.py
@@ -46,7 +46,7 @@ def __init__(
             dataset_archive = DEFAULT_DATA_ARCHIVE
 
         data_path = tf.keras.utils.get_file(
-            origin=DEFAULT_DATA_ARCHIVE,
+            origin=dataset_archive,
             untar=True,
         )