diff --git a/tests/test_models.py b/tests/test_models.py
index 570b49db..c1632271 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -32,7 +32,7 @@ else:
 
 TARGET_FWD_SIZE = MAX_FWD_SIZE = 384
 TARGET_BWD_SIZE = 128
-MAX_BWD_SIZE = 384
+MAX_BWD_SIZE = 320
 MAX_FWD_FEAT_SIZE = 448