fixed test cases
small minor fixes