Fixed output issue