I don't know if you have read Marx's Critique of the Gotha Programme, but if not I want to quote this part that feels relevant to what your saying. Mainly just that even under a socialist state, the value produced by the worker isn't gonna 100% belong to them.

Do not the bourgeois assert that the present-day distribution is "fair"? And is it not, in fact, the only "fair" distribution on the basis of the present-day mode of production? Are economic relations regulated by legal conceptions, or do not, on the contrary, legal relations arise out of economic ones? Have not also the socialist sectarians the most varied notions about "fair" distribution?

To understand what is implied in this connection by the phrase "fair distribution", we must take the first paragraph and this one together. The latter presupposes a society wherein the instruments of labor are common property and the total labor is co-operatively regulated, and from the first paragraph we learn that "the proceeds of labor belong undiminished with equal right to all members of society."

"To all members of society"? To those who do not work as well? What remains then of the "undiminished" proceeds of labor? Only to those members of society who work? What remains then of the "equal right" of all members of society?

But "all members of society" and "equal right" are obviously mere phrases. The kernel consists in this, that in this communist society every worker must receive the "undiminished" Lassallean "proceeds of labor".

Let us take, first of all, the words "proceeds of labor" in the sense of the product of labor; then the co-operative proceeds of labor are the total social product.

From this must now be deducted: First, cover for replacement of the means of production used up. Second, additional portion for expansion of production. Third, reserve or insurance funds to provide against accidents, dislocations caused by natural calamities, etc.

These deductions from the "undiminished" proceeds of labor are an economic necessity, and their magnitude is to be determined according to available means and forces, and partly by computation of probabilities, but they are in no way calculable by equity.

There remains the other part of the total product, intended to serve as means of consumption.

Before this is divided among the individuals, there has to be deducted again, from it: First, the general costs of administration not belonging to production. This part will, from the outset, be very considerably restricted in comparison with present-day society, and it diminishes in proportion as the new society develops. Second, that which is intended for the common satisfaction of needs, such as schools, health services, etc. From the outset, this part grows considerably in comparison with present-day society, and it grows in proportion as the new society develops. Third, funds for those unable to work, etc., in short, for what is included under so-called official poor relief today.

Only now do we come to the "distribution" which the program, under Lassallean influence, alone has in view in its narrow fashion – namely, to that part of the means of consumption which is divided among the individual producers of the co-operative society.

The "undiminished" proceeds of labor have already unnoticeably become converted into the "diminished" proceeds, although what the producer is deprived of in his capacity as a private individual benefits him directly or indirectly in his capacity as a member of society.

Just as the phrase of the "undiminished" proceeds of labor has disappeared, so now does the phrase of the "proceeds of labor" disappear altogether.

Within the co-operative society based on common ownership of the means of production, the producers do not exchange their products; just as little does the labor employed on the products appear here as the value of these products, as a material quality possessed by them, since now, in contrast to capitalist society, individual labor no longer exists in an indirect fashion but directly as a component part of total labor. The phrase "proceeds of labor", objectionable also today on account of its ambiguity, thus loses all meaning.

Critique of the Gotha Programme