Fix the things I forgot