[TESTS] Adding tests to improve coverage.