CHANGE: also keep reconstruction information