Fix potential bug.