Loading matplotlib pickle figure in a tkinter frame

I have a small interface which loads data into a matplotlib figure. I would like to be able to load pickles which contain previously made plots but somehow I cannot make it work.

This is one of the examples from matplotlib to embed a plot within tkinter:

import sys
import pickle
import matplotlib
from numpy import arange, sin, pi, cos
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2TkAgg
from matplotlib.backend_bases import key_press_handler
from matplotlib.figure import Figure
if sys.version_info[0] < 3:
    import Tkinter as Tk
else:
    import tkinter as Tk

def restore_from_pickle(pickle_address, fig):
    fig.clear()
    with open(pickle_address, 'rb') as fid:
        fig = pickle.load(fid)
    return

def _quit():
    root.quit()
    root.destroy()

def on_key_event(event):
    print('you pressed %s' % event.key)
    key_press_handler(event, canvas, toolbar)

matplotlib.use('TkAgg')
root = Tk.Tk()
root.wm_title("Embedding in TK")

f = Figure(figsize=(5, 4), dpi=100)
t = arange(0.0, 3.0, 0.01)

#Plot to store
a = f.add_subplot(111)
s = sin(2*pi*t)
a.plot(t, s)

#Saving to pickle
with open('figure_pickle', 'wb') as fid:
    pickle.dump(f, fid)

#Initial plot display
f.clear()
a = f.add_subplot(111)
s = cos(2*pi*t)
a.plot(t, s)

# a tk.DrawingArea
canvas = FigureCanvasTkAgg(f, master=root)
canvas.show()
canvas.get_tk_widget().pack(side=Tk.TOP, fill=Tk.BOTH, expand=1)

toolbar = NavigationToolbar2TkAgg(canvas, root)
toolbar.update()
canvas._tkcanvas.pack(side=Tk.TOP, fill=Tk.BOTH, expand=1)

canvas.mpl_connect('key_press_event', on_key_event)

button = Tk.Button(master=root, text='f(sin)', command=restore_from_pickle('figure_pickle', f))
#button = Tk.Button(master=root, text='Quit', command=_quit)
button.pack(side=Tk.BOTTOM)

Tk.mainloop() 

This produces something like:

If you press the button it closes the window. I modified it so it saves a pickle containing one figure and the button now loads this image. However, when I run it:

The original plot is not drawn and neither the one in the pickle...

What is happening?

Thanks for any advice