If revolution happens within our lifetimes, there's no way a socialist state outlives its usefulness on that timescale, in the best case scenario where nothing goes wrong and capitalism is extinguished everywhere. The dissolution of the class distinctions and withering away of the state would have to happen over the course of many generations. It's not until the revolutionary generation is long-dead that the nonexistence of "sufficiently motivated assholes" is ensured.