Fix wheel build

@cournape Fix wheel build.