change restore method signature, see #56