Revert "fix test"

This reverts commit 901e0a9e92d6b84e3907e1a2e9307debd0259a61.