Fix a wobbly Shimano freehub?
If its brand new and under warranty, don't mess around with it... take it back.

Your freehub should have pretty much no discernible play, and adding a cassette on top just makes it look worse.

As for fixing the freehub yourself, I don't know because I've not yet had one that needs repair.