This is done by using `preserve_metadata` instead of a custom `after_dig_node` callback.
Further balancing or adjustments might be necessary.