[TESTS] Improve coverage (#5)