#37 Bug fix when editting assignment