[TESTS] Fixing incoherency in unit tests