fixes the issue where the dropdown menu would start with the wrong stage value when you edit