fix custom run feature