How to group fields in the form without grouping them in the node view?
Found it. Turns out the array keys are very important. The first time around, I added the form fields to the fieldset without any keys. It started working just by adding the proper array keys to the fieldset, see the last revision.