partial fix restore action,  to what it originally supposes to be.

Signed-off-by: Ace Han <ace.jl.han@gmail.com>