revert "Fix test"

This reverts commit aae997f226821aa5515b5c72e5138198896a8a13.