Fix for install from pypi

Fix for install from pypi