Corrected a small error in one of the tests