AT Arm should work fine - as long as you get into FLC and have speeds set to Auto. If your speeds are on manual or somethings wrong with your FMS speeds, then the FLC mode might hold VY all the way up which is way too steep for a comfortable level off.
Also verify that the auto throttle brought the throttles to climb mode and not still in to/ga setting.