Fixed bug with only getting one cluster per stage