Swap deprecated classes in test elements