Fix bits I missed