Minor Update

* Last commit missed a couple of changes