Fix potential bug: add assumption invalidation