Fix for python environment