Update testing code.