Revert "Fix test"

This reverts commit f392d95ac34361ac9b38f9f39baf57500717e665.