Fixed a bug where the edit widget code was not being persisted within the client storage