fix for stage helper