Fix nested stages by adding missing message stage clear param