Fix mistake introduced by previous commit