Fix pb on tests