diff --git a/examples/hed/datasets/get_dataset.py b/examples/hed/datasets/get_dataset.py
index 5d02b0c..cea07bc 100644
--- a/examples/hed/datasets/get_dataset.py
+++ b/examples/hed/datasets/get_dataset.py
@@ -59,8 +59,8 @@ def split_equation(equations_by_len, prop_train, prop_val):
     """
     Split the equations in each length to training and validation data according to the proportion
     """
-    train_equations_by_len = {1: dict(), 0: dict()}
-    val_equations_by_len = {1: dict(), 0: dict()}
+    train_equations_by_len = {1: {}, 0: {}}
+    val_equations_by_len = {1: {}, 0: {}}
 
     for label in range(2):
         for equation_len, equations in equations_by_len[label].items():
@@ -80,7 +80,7 @@ def get_dataset(dataset="mnist", train=True):
 
     if not os.path.exists(data_dir):
         print("Dataset not exist, downloading it...")
-        url = "https://drive.google.com/u/0/uc?id=1XoJDjO3cNUdytqVgXUKOBe9dOcUBobom&export=download"
+        url = "https://drive.google.com/u/0/uc?id=1W2AUn_fnXa4XkgLk4d17K3bEgpae8GMg&export=download"
         download_and_unzip(url, os.path.join(CURRENT_DIR, "HED.zip"))
         print("Download and extraction complete.")
 
diff --git a/examples/hwf/datasets/get_dataset.py b/examples/hwf/datasets/get_dataset.py
index f8ce374..0700d3c 100644
--- a/examples/hwf/datasets/get_dataset.py
+++ b/examples/hwf/datasets/get_dataset.py
@@ -31,7 +31,7 @@ def get_dataset(train=True, get_pseudo_label=False):
 
     if not os.path.exists(data_dir):
         print("Dataset not exist, downloading it...")
-        url = "https://drive.google.com/u/0/uc?id=1G07kw-wK-rqbg_85tuB7FNfA49q8lvoy&export=download"
+        url = "https://drive.google.com/u/0/uc?id=1t52OE2Wdm5GdShX1jD2Wy8phCllk0r8I&export=download"
         download_and_unzip(url, os.path.join(CURRENT_DIR, "HWF.zip"))
         print("Download and extraction complete.")