Fixed missing line of code