Warrington Wolves move back to the top of the Super League table as they come from behind to beat St Helens.