Minor update for P2 feedback. Missed a line