Fix integration tests broken in f44b59a