[test/unit] Fix bad refactoring