Revert "fix test"

This reverts commit 85c390ab6315f097d469745c5a93ec8a9a571b36.