Move snapshot set state in remove_snapshot