Small adjustment for python-mode