Fix adding to the stages oops