Revert "fix test"

This reverts commit b1b7b6eca2a5965d72554047021b917ae09cde90.