Fix restore command