Add missing change from the corresponding float branch