Missed a few minor changes