Fix some crappy code.