Fix integration tests (#40)