fix jedi DeprecationWarning