Roadworks are set to hit a major A-road in Wiltshire after the area was flooded.