Fix a missing change in last commit