Revert "Fix test fail"

This reverts commit 1de0f85c7373e7e0b9ec0532f237d6320bb5c047.