fixed test cases that were failing