Fix to clean up

Missed a change