Add code missing in the preparation of the previous commit