Fix pip install issue