Revert "Fix failing test"

This reverts commit 8a5d5afcf09c414e560303d8b340e1e62bda7b25.