Update testing helpers