Add missing line of code