fix: missing code button, fix: the code was lost when editing