Missed a couple of silly type annotations