Output shaft assembly