fix remove/restore