small tests fix