Updates Test Cases
Update tests
Revert "Update tests"

This reverts commit ab186693c6cd3dd3c59d027ee4f4b84def56ab4b.
Modify test cases