Fixed output problem