It works as you would expect in MTW (1), in STW it does not. To get unit stacks back into the castle in STW you need to first select the stack so that it appears in the review panel at the bottom of the screen, then select every individual unit (or whichever units you want) and drag and drop them over the castle.