Fix some errors I missed