updated test cases to pass with new change