Missed small but important part of previous update