diff --git a/swords/datasets.py b/swords/datasets.py
index d7d6f8f..75298d3 100644
--- a/swords/datasets.py
+++ b/swords/datasets.py
@@ -1501,9 +1501,9 @@ def _print(x):
     # Load from disk
     _print('Loading from {dataset}')
     if dataset.endswith('.gz'):
-      open_fn = open
-    else:
       open_fn = gzip.open
+    else:
+      open_fn = open
     with open_fn(dataset, 'r') as f:
       d = LexSubDataset.from_dict(json.load(f))
   else: