Ugly test fixes