Fix: three lines missed in the last commit