fixed bug with removing stages