diff --git a/neuralogic/data.py b/neuralogic/data.py
index c0f46239..1941130f 100644
--- a/neuralogic/data.py
+++ b/neuralogic/data.py
@@ -127,4 +127,5 @@ def validate_query(rule: str) -> bool:
 base_path = os.path.abspath(os.path.dirname(__file__))
 
 XOR = Dataset(source_dir=os.path.join(base_path, "..", "dataset", "simple", "xor", "naive"))
+XOR_Vectorized = Dataset(source_dir=os.path.join(base_path, "..", "dataset", "simple", "xor", "vectorized"))
 Mutagenesis = Dataset(source_dir=os.path.join(base_path, "..", "dataset", "molecules", "mutagenesis"))