Make it work with IPython v1.1.0.