Make tests compatible with totem-run