Update test cases for Logic part