Add change missing from previous commit