fixed issue with jedi importing