Adapt to collect stage information