And by that I mean that each layer with sublayers in AI should become a group in PS.
Check 'Write Layers'