diff --git a/tests/test_models.py b/tests/test_models.py
index 570b49db54e9be6dc39349edcc78557e21295800..c1632271e456e14ce86ea3838813cb77afc18ebe 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