Tiny tweak

Missed a line or two