Fix a missing part of the code from merging.