Fixed Python integration tests