Update to work with newest version of jedi plugin