Fix warning restore