Missed a couple of changes for the last commit.