That's where the permanent revolution would help. The workers must not allow splitting the revolution into stages of concessions and compromises but fight until total victory and the dissolution of the state.

Also this is the reason why communists are not pacifists — the working class has the right and a duty to defend itself and it's gains. That's what Marx meant when he wrote that under no pretense must workers be disarmed.