Fix one of few tests :)