Update after integration tests