Fix for python package setup