Revert "Fixed failing tests"

This reverts commit 373ec114f6e3bfb523a2ad3022170aa213334d90.
fixed failing test