diff --git a/data/transform/check_sequence_length_transform.py b/data/transform/check_sequence_length_transform.py
index 040737af8acd40491939b62def17a4bd7be584e6..961610ad867f8009ae1e25865efdce1ec2039974 100644
--- a/data/transform/check_sequence_length_transform.py
+++ b/data/transform/check_sequence_length_transform.py
@@ -13,9 +13,9 @@ class CheckSequenceLenghtTransform():
         'double_substitution': 2,  # invalid single module embedding
         'composite': [0, 0, 0, 0, 0, 1, 2, 2],
         'composite_A': [0, 0, 0, 0, 0, 1, 2, 2],
-        'composite_B': [0, 0, 0, 0, 0, 1, 2, 2],
-        'composite_C': [0, 0, 0, 0, 1, 2],
-        'composite_D': [0, 0, 0, 0, 1, 2, 2, 2],
+        'composite_B': [0, 0, 0, 0, 0, 1],
+        'composite_C': [0, 0, 0, 0, 0, 1],
+        'composite_D': [0, 0, 0, 0, 1, 1, 2, 2],
     }
 
     _convolution_factor_map = {
@@ -30,9 +30,9 @@ class CheckSequenceLenghtTransform():
         'double_substitution': 8,
         'composite': [1, 1, 1, 4, 8, 8, 8, 8],
         'composite_A': [1, 1, 1, 4, 8, 8, 8, 8],
-        'composite_B': [1, 1, 1, 2, 8, 8, 8, 8],
-        'composite_C': [1, 1, 4, 8, 8, 8],
-        'composite_D': [1, 1, 2, 8, 4, 2, 4, 8],
+        'composite_B': [1, 1, 1, 1, 8, 8],
+        'composite_C': [1, 1, 2, 4, 8, 4],
+        'composite_D': [1, 1, 4, 8, 4, 8, 4, 8],
     }
 
     def __init__(self, num_positions, embedding):