fixed broken test cases