Revert "Add more cases for test"

This reverts commit 06e15070378830252cdcec3841e47b9ff6ec9058.