Revert "Fixed a test error"

This reverts commit 718e7cbb28a2909bfdcc74ec5da832922afcebe0.
Revert "Revert "Fixed a test error""

This reverts commit 9921fce956aed0b2f81696bf7b49a2ecbba52429.