Fix pip package