Update the correct class