diff --git a/combo/data/dataset.py b/combo/data/dataset.py
index 870659fe27857d157e5061f77690f61f166330a7..29b1f918c5de55f8055ebb432d3fc90e97ba2c42 100644
--- a/combo/data/dataset.py
+++ b/combo/data/dataset.py
@@ -204,7 +204,7 @@ class UniversalDependenciesDatasetReader(allen_data.DatasetReader):
             default_value = [0.0] * classes_count
             padded_tags = util.pad_sequence_to_length(field._indexed_multi_labels, desired_num_tokens,
                                                       lambda: default_value)
-            tensor = torch.LongTensor(padded_tags)
+            tensor = torch.tensor(padded_tags, dtype=torch.long)
             return tensor
 
         return as_tensor