ENH: make it work with IPython 2.0dev

I do not know what is the status of this project... but this change make it work in the current master branch of IPython. (2.0.dev)
At least for a simple usage of the extension...