Fix some python imports