diff --git a/tests/icosagon/test_declayer.py b/tests/icosagon/test_declayer.py
index 732fd45..014d304 100644
--- a/tests/icosagon/test_declayer.py
+++ b/tests/icosagon/test_declayer.py
@@ -166,7 +166,7 @@ def test_decode_layer_05():
     # print(rel_pred.edges_neg.train)
 
     repr_in = in_layer(None)
-    assert isinstance(repr_in, list)
+    assert isinstance(repr_in, torch.nn.ParameterList)
     assert len(repr_in) == 1
     assert isinstance(repr_in[0], torch.Tensor)
     assert torch.all(repr_in[0].to_dense() == torch.eye(10))