Fixing broken code that was weird for some reason