Fix ugly hack.