Attempt to fix to work with pip