revert fix, just fix the test