diff --git a/test/test_crosslink_input.py b/test/test_crosslink_input.py
index de31716b..4803d8fb 100644
--- a/test/test_crosslink_input.py
+++ b/test/test_crosslink_input.py
@@ -7,7 +7,7 @@
 class TestCreateObjects(unittest.TestCase):
     def setUp(self) -> None:
         self.crosslink_info ="./test/test_data/test_xl_input.pkl.gz"
-        self.asym_id = [1]*10 + [2]*25 + [3]*40
+        self.asym_id = torch.tensor([1]*10 + [2]*25 + [3]*40)
         self.chain_id_map = {
             "A":_FastaChain(sequence='',description='chain1'),
             "B":_FastaChain(sequence='',description='chain2'),