Fixing some silly things I forgot