Fix for 'can't edit steps with edit mode' bug