Fixed ugly test