Revert "Fix test"

This reverts commit 5f3509b266786cc4a591ac09314fc7e03aac1b2b.