Fixing deprecation warning with IPython 2.0