Update testing patch to work correctly.