Fixing some errors I missed