diff --git a/go_ia/main.py b/go_ia/main.py
index 146bee2..83130f2 100755
--- a/go_ia/main.py
+++ b/go_ia/main.py
@@ -256,6 +256,7 @@ def setup_device():
 
 
 if __name__ == "__main__":
+    torch.serialization.add_safe_globals([GoDataset])
     parser = argparse.ArgumentParser(
         prog="alphaChadGo", description="a random ai evaluating go", epilog=""
     )