Maintain correct state when going back and editing old answers