Fix small omission in last commit