Bagnaia back on form