Closed-loop shaft control