Fix note edit functionality