Ugly hack fix, lol.