fixed test cases that were not ending soon enough