Bug fix cont

Missed a spot