Fixed display error on saving