diff --git a/dataset/shapenet.py b/dataset/shapenet.py
index 6d6cd57..7f0171a 100644
--- a/dataset/shapenet.py
+++ b/dataset/shapenet.py
@@ -24,6 +24,7 @@ class CustomShapeNet(InMemoryDataset):
 
     def __init__(self, root, collate_per_segment=True, train=True, transform=None, pre_filter=None, pre_transform=None,
                  headers=True, has_variations=False):
+        assert not (collate_per_segment and has_variations), 'Either use each element or pointclouds - with variations'
         self.has_headers = headers
         self.has_variations = has_variations
         self.collate_per_element = collate_per_segment