Add accidentally removed code