fix tests
Revert "fix tests"

This reverts commit fbd51b77a047f2a04a94c4b474881c2be9bf1171.