Missed a conditional check, added.