Fix tree toggles in VM tests