Fix failing integration tests for the wheel client