Acidentally missed a part in the last commit