diff --git a/crabnet/model.py b/crabnet/model.py
index 8842e82..6403a47 100644
--- a/crabnet/model.py
+++ b/crabnet/model.py
@@ -72,9 +72,6 @@ def load_data(self, file_name, batch_size=2**9, train=False):
         print(f'loading data with up to {data_loaders.n_elements:0.0f} '
               f'elements in the formula')
 
-        # update n_elements after loading dataset
-        self.n_elements = data_loaders.n_elements
-
         data_loader = data_loaders.get_data_loaders(inference=inference)
         y = data_loader.dataset.data[1]
         if train: