Fix for pip-based setup