Added missing line of code