Update snippet for test