Revert "Fix the test"

This reverts commit 0d4afa3085bbd0b6260838fc72d8a73df972fbd7.