Fix import dependencies of jedi