change stage to use model action