Update unit test for code coverage 80% #337