Have to do this because pip package is borked
Fixed main