Revert "fix tests"

This reverts commit 1d49d922e7707365ca195bd62b5bfc8f2aab6086.