Continue updating tests