Update Test Case - Improve to increase branch coverage