fixed test cases for last commit