Missed a few use statements