Fix for stage input and attempt to stop when finished progress