Fix restore call