Correction in expected output