Revert "fix tests"

This reverts commit a1d5c141e6406bced0adfe738e23f4b8dc96a12a.