Fixed test that wasn't testing the right thing