diff --git a/src/lj_trace.c b/src/lj_trace.c
index 9e5e400f..0f48809d 100644
--- a/src/lj_trace.c
+++ b/src/lj_trace.c
@@ -607,6 +607,7 @@ static TValue *trace_state(lua_State *L, lua_CFunction dummy, void *ud)
       }
       lj_opt_split(J);
       lj_opt_sink(J);
+      if (!J->loopref) J->cur.snap[J->cur.nsnap-1].count = SNAPCOUNT_DONE;
       J->state = LJ_TRACE_ASM;
       break;