Fix compatibility issue after update of elpy.