Fixed Integration tests

Thanks @heynemann