Revert "fix unit test"

This reverts commit 42a2548200ed84ca4aad7ec5ae3398d210c0e491.