Fix assignment editing