Corrected error in tests