Don't leak implicit return from #dispatch into #assume_state