diff --git a/src/tbf_ul_ass_fsm.c b/src/tbf_ul_ass_fsm.c
index eab34ee..618e072 100644
--- a/src/tbf_ul_ass_fsm.c
+++ b/src/tbf_ul_ass_fsm.c
@@ -257,7 +257,9 @@
 	},
 	[TBF_UL_ASS_SEND_ASS] = {
 		.in_event_mask = X(TBF_UL_ASS_EV_CREATE_RLCMAC_MSG),
-		.out_state_mask = X(TBF_UL_ASS_WAIT_ACK),
+		.out_state_mask =
+			X(TBF_UL_ASS_WAIT_ACK) |
+			X(TBF_UL_ASS_NONE),
 		.name = "SEND_ASS",
 		.action = st_send_ass,
 	},