fix adding stages