Minor Error correction in tests