Go with color harmony that is beneficial to the baby.

Baby is posed on dark background replace it with lighter color.

Same goes for bed sheet.

Show the baby and don't show the floor.

There is dark shadow of the soft box keep that out to the image.

Move the baby's left hand and put it under the cheek.

That will improve the composition.