Fix problem with refreshing the edit widget