Fixed issue preventing use after pip install