Fix some testing change that I forget to revert