Missed code in previous update