Missed to update some pasted code