Tony Stewart back on track at AMS