Small fix of something I broke earlier.