Make the test the same in both places